Spaces:
Runtime error
Runtime error
Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -99,7 +99,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
99 |
tokenized_test = test_dataset.map(preprocess_function, batched=True)
|
100 |
|
101 |
## Modify inference model
|
102 |
-
training_args = torch.load("./tasks/utils/training_args.bin")
|
103 |
training_args.eval_strategy='no'
|
104 |
|
105 |
trainer = Trainer(
|
|
|
99 |
tokenized_test = test_dataset.map(preprocess_function, batched=True)
|
100 |
|
101 |
## Modify inference model
|
102 |
+
training_args = torch.load("./tasks/utils/training_args.bin",weights_only = False)
|
103 |
training_args.eval_strategy='no'
|
104 |
|
105 |
trainer = Trainer(
|