Spaces:

Baleegh
/

Fluency_Score

Runtime error

Abdulmohsena commited on 16 days ago

Commit

4d4f55e

verified ·

1 Parent(s): 150bdae

Update Fluency_Score.py

Files changed (1) hide show

Fluency_Score.py CHANGED Viewed

@@ -40,7 +40,7 @@ class Fluency_Score(evaluate.Measurement):
             max_length=128
         ).to(device)
-        with torch.no_grad():
             output = self.model(**inputs)
             prediction = output.logits.clip(0, 1)

             max_length=128
         ).to(device)
+        with torch.inference_mode():
             output = self.model(**inputs)
             prediction = output.logits.clip(0, 1)