Fix format for filepaths
Browse files- run_mlm_flax_stream.py +1 -1
run_mlm_flax_stream.py
CHANGED
@@ -402,7 +402,7 @@ if __name__ == "__main__":
|
|
402 |
boundaries=sampling_args.boundaries,
|
403 |
perplexity_model=sampling_args.perplexity_model,
|
404 |
seed=training_args.seed,
|
405 |
-
filepaths=filepaths,
|
406 |
)
|
407 |
|
408 |
if model_args.config_name:
|
|
|
402 |
boundaries=sampling_args.boundaries,
|
403 |
perplexity_model=sampling_args.perplexity_model,
|
404 |
seed=training_args.seed,
|
405 |
+
filepaths={"train": filepaths},
|
406 |
)
|
407 |
|
408 |
if model_args.config_name:
|