Replaced deepseek with qwq
This commit is contained in:
15
tools/train/bin/train
Normal file
15
tools/train/bin/train
Normal file
@@ -0,0 +1,15 @@
|
||||
#!/bin/bash
|
||||
|
||||
set -e
|
||||
|
||||
SIA_DIR="/root/sia"
|
||||
OUTPUT_DIR="${1:-/root/models/$(cd "$SIA_DIR" && git rev-parse HEAD)}"
|
||||
|
||||
if [ -n "$(cd "$SIA_DIR" && git status --porcelain)" ]; then
|
||||
echo "Uncommitted changes in SIA directory"
|
||||
#exit 1
|
||||
fi
|
||||
|
||||
mkdir -p "$OUTPUT_DIR"
|
||||
|
||||
python -m train.qwq "$@"
|
||||
@@ -1,10 +0,0 @@
|
||||
#!/root/venvs/train/bin/python
|
||||
"""
|
||||
Command-line utility for fine-tuning DeepSeek models using Unsloth.
|
||||
Always trains from a base model to create a new fine-tuned model.
|
||||
"""
|
||||
import sys
|
||||
from train.unsloth_deepseek import main
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(main())
|
||||
@@ -1,9 +0,0 @@
|
||||
#!/root/venvs/train/bin/python
|
||||
"""
|
||||
Command-line utility for fine-tuning Mistral models using Mistral API.
|
||||
"""
|
||||
import sys
|
||||
from train.mistral_api import main
|
||||
|
||||
if __name__ == "__main__":
|
||||
sys.exit(main())
|
||||
Reference in New Issue
Block a user