Spaces:
Sleeping
Sleeping
Upload run_cloud_training.py with huggingface_hub
Browse files- 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
|
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
|