ASTERIZER commited on
Commit
076e1b4
·
verified ·
1 Parent(s): ef22d93

Upload gpu_train.sh with huggingface_hub

Browse files
Files changed (1) hide show
  1. gpu_train.sh +9 -0
gpu_train.sh CHANGED
@@ -110,4 +110,13 @@ CUDA_VISIBLE_DEVICES=0 python3 lora_sft_train.py \
110
  echo "============================================================"
111
  echo " Training complete!"
112
  echo " Adapter saved to: Base/out/sft/rag_mcp_lora/final/"
 
 
 
 
 
 
 
 
 
113
  echo "============================================================"
 
110
  echo "============================================================"
111
  echo " Training complete!"
112
  echo " Adapter saved to: Base/out/sft/rag_mcp_lora/final/"
113
+ echo " To upload it to Hugging Face, run:"
114
+ echo " python3 upload_lora_to_hf.py --repo-id ASTERIZER/LUNA-100M --folder Base/out/sft/rag_mcp_lora/final --path-in-repo rag_mcp_lora/final"
115
+ if [ "${UPLOAD_TO_HF:-0}" = "1" ]; then
116
+ echo " UPLOAD_TO_HF=1 detected. Uploading adapter to Hugging Face..."
117
+ python3 upload_lora_to_hf.py \
118
+ --repo-id ASTERIZER/LUNA-100M \
119
+ --folder Base/out/sft/rag_mcp_lora/final \
120
+ --path-in-repo rag_mcp_lora/final
121
+ fi
122
  echo "============================================================"