Joshua Lochner commited on
Commit
87b2dec
·
1 Parent(s): 721bf64

Fix `--no_cuda` argument for preprocessing

Browse files
Files changed (1) hide show
  1. src/preprocess.py +2 -2
src/preprocess.py CHANGED
@@ -539,7 +539,7 @@ def main():
539
  ModelArguments,
540
  GeneralArguments
541
  ))
542
- preprocess_args, dataset_args, segmentation_args, model_args, _ = hf_parser.parse_args_into_dataclasses()
543
 
544
  raw_dataset_path = os.path.join(
545
  preprocess_args.raw_data_dir, preprocess_args.raw_data_file)
@@ -796,7 +796,7 @@ def main():
796
 
797
  from model import get_model_tokenizer
798
  model, tokenizer = get_model_tokenizer(
799
- model_args.model_name_or_path, model_args.cache_dir, model_args.no_cuda)
800
 
801
  # TODO
802
  # count_videos = 0
 
539
  ModelArguments,
540
  GeneralArguments
541
  ))
542
+ preprocess_args, dataset_args, segmentation_args, model_args, general_args = hf_parser.parse_args_into_dataclasses()
543
 
544
  raw_dataset_path = os.path.join(
545
  preprocess_args.raw_data_dir, preprocess_args.raw_data_file)
 
796
 
797
  from model import get_model_tokenizer
798
  model, tokenizer = get_model_tokenizer(
799
+ model_args.model_name_or_path, model_args.cache_dir, general_args.no_cuda)
800
 
801
  # TODO
802
  # count_videos = 0