Spaces:
Sleeping
Sleeping
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,
|