"tests/models/vscode:/vscode.git/clone" did not exist on "722bf7efcce72e60412f75d6775af7b03041d8c8"
  • Matthew Hoffman's avatar
    Fix TrainingArguments regression with torch <2.0.0 for dataloader_prefetch_factor (#29447) · 2890116a
    Matthew Hoffman authored
    * Fix TrainingArguments regression with torch <2.0.0 for dataloader_prefetch_factor
    
    dataloader_prefetch_factor was added to TrainingArguments in #28498 with the default value None, but  versions of torch<2.0.0 do not accept None and will raise an error if num_workers == 0 and prefetch_factor != 2
    
    * Add is_torch_available() check
    
    * Use is_torch_greater_or_equal_than_2_0
    
    add back check for dataloader_prefetch_factor
    2890116a
training_args.py 138 KB