TheoLvs commited on
Commit
5d388ca
·
verified ·
1 Parent(s): ab0a673

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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(