MatthiasPicard commited on
Commit
5070ea4
·
verified ·
1 Parent(s): 91c1ff4

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. tasks/text.py +4 -1
tasks/text.py CHANGED
@@ -77,7 +77,10 @@ async def evaluate_text(request: TextEvaluationRequest):
77
 
78
  # training_args = torch.load("training_args.bin")
79
  # training_args.eval_strategy='no'
80
-
 
 
 
81
  trainer = Trainer(
82
  model=model,
83
  # args=training_args,
 
77
 
78
  # training_args = torch.load("training_args.bin")
79
  # training_args.eval_strategy='no'
80
+
81
+ model = model.half()
82
+ model.eval()
83
+
84
  trainer = Trainer(
85
  model=model,
86
  # args=training_args,