feat: unload AI model before training
This commit is contained in:
@@ -10,6 +10,11 @@ echo "=== LoRA Training Setup ==="
|
|||||||
echo "Repo: ${REPO_DIR}"
|
echo "Repo: ${REPO_DIR}"
|
||||||
echo ""
|
echo ""
|
||||||
|
|
||||||
|
# Unload AI model to free VRAM
|
||||||
|
echo "=== Unloading AI model ==="
|
||||||
|
ai none
|
||||||
|
sleep 2
|
||||||
|
|
||||||
# Check Python
|
# Check Python
|
||||||
if ! command -v python3 &> /dev/null; then
|
if ! command -v python3 &> /dev/null; then
|
||||||
echo "ERROR: python3 not found"
|
echo "ERROR: python3 not found"
|
||||||
|
|||||||
Reference in New Issue
Block a user