PEFT
qlora
sft
trl
qwen3
tmf921
intent-based-networking
network-slicing
rtx-6000-ada
ml-intern
nraptisss commited on
Commit
0187cea
·
verified ·
1 Parent(s): 3e07f1e

Fix stage2 push_to_hub dataset metadata validation

Browse files
Files changed (1) hide show
  1. scripts/train_continue_adapter.py +5 -1
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
- trainer.push_to_hub(commit_message="Stage2 weak-layer QLoRA continuation", dataset_name="weak-layer replay local parquet")
 
 
 
 
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