Replaced deepseek with qwq

This commit is contained in:
Niels Geens
2025-03-14 11:22:30 +01:00
parent 3ea3239a9b
commit 2e66020f8e
13 changed files with 693 additions and 530 deletions

15
tools/train/bin/train Normal file
View 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 "$@"

View File

@@ -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())

View File

@@ -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())