Fix stage2 push_to_hub dataset metadata validation
Browse files
scripts/train_continue_adapter.py
CHANGED
|
@@ -199,7 +199,11 @@ def main():
|
|
| 199 |
tokenizer.save_pretrained(cfg["output_dir"])
|
| 200 |
|
| 201 |
if bool(cfg.get("push_to_hub", True)):
|
| 202 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 203 |
print(f"Pushed stage2 adapter to https://huggingface.co/{cfg.get('hub_model_id')}")
|
| 204 |
|
| 205 |
|
|
|
|
| 199 |
tokenizer.save_pretrained(cfg["output_dir"])
|
| 200 |
|
| 201 |
if bool(cfg.get("push_to_hub", True)):
|
| 202 |
+
# dataset_name must be a valid HF dataset id for model-card metadata validation.
|
| 203 |
+
trainer.push_to_hub(
|
| 204 |
+
commit_message="Stage2 weak-layer QLoRA continuation",
|
| 205 |
+
dataset_name="nraptisss/TMF921-intent-to-config-research-sota",
|
| 206 |
+
)
|
| 207 |
print(f"Pushed stage2 adapter to https://huggingface.co/{cfg.get('hub_model_id')}")
|
| 208 |
|
| 209 |
|