George-API commited on
Commit
0cd991c
·
verified ·
1 Parent(s): 1faea13

Upload run_cloud_training.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. run_cloud_training.py +1 -3
run_cloud_training.py CHANGED
@@ -626,9 +626,7 @@ def train(config_path, dataset_name, output_dir, upload_to_hub=False, hub_repo_n
626
  ),
627
  # Add label_names to avoid warning
628
  compute_metrics=None,
629
- tokenizer=tokenizer, # Provide tokenizer for proper padding
630
- # Define label_names to fix warning
631
- label_names=["labels"]
632
  )
633
 
634
  # Start training
 
626
  ),
627
  # Add label_names to avoid warning
628
  compute_metrics=None,
629
+ tokenizer=tokenizer # Provide tokenizer for proper padding
 
 
630
  )
631
 
632
  # Start training