Upload gpu_train.sh with huggingface_hub
Browse files- 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 "============================================================"
|