versae commited on
Commit
7d6bbb2
1 Parent(s): 13757a8

Fix format for filepaths

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