Spaces:
Running
Running
Joshua Lochner
commited on
Commit
·
87b2dec
1
Parent(s):
721bf64
Fix `--no_cuda` argument for preprocessing
Browse files- 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,
|
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,
|
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
|