micro zapusk no cross

50e0b2f7 · Мазур Грета Евгеньевна · c064e465 · 50e0b2f7 · 50e0b2f7
Commit 50e0b2f7 authored 3 weeks ago by Мазур Грета Евгеньевна
Hide whitespace changes
Inline Side-by-side

Showing

with 4 additions and 4 deletions
+4 -4
--- a/.ipynb_checkpoints/micro_no_cross-checkpoint.py
+++ b/.ipynb_checkpoints/micro_no_cross-checkpoint.py
@@ -203,8 +203,8 @@ def predict(text):
    
    with torch.no_grad():
        outputs = model(**inputs)
-        safety_probs = torch.softmax(outputs['safety'], dim=1)
-        attack_probs = torch.softmax(outputs['attack'], dim=1)
+        safety_probs = torch.softmax(outputs['logits_safety'], dim=1)
+        attack_probs = torch.softmax(outputs['logits_attack'], dim=1)
    
    return {
        'safety': {

--- a/micro_no_cross.py
+++ b/micro_no_cross.py
@@ -203,8 +203,8 @@ def predict(text):
    
    with torch.no_grad():
        outputs = model(**inputs)
-        safety_probs = torch.softmax(outputs['safety'], dim=1)
-        attack_probs = torch.softmax(outputs['attack'], dim=1)
+        safety_probs = torch.softmax(outputs['logits_safety'], dim=1)
+        attack_probs = torch.softmax(outputs['logits_attack'], dim=1)
    
    return {
        'safety': {