Update tasks/text.py
Browse files- 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,
|