#!/bin/bash
cd ~/sd/kohya-sd-scripts
# ./venv/bin/pip install --upgrade pip
# ./venv/bin/pip install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu
# ./venv/bin/pip install --upgrade xformers

# ./venv/bin/python networks/extract_lora_from_models.py -h
./venv/bin/python networks/extract_lora_from_models.py \
--model_org ~/sd/ComfyUI/models/checkpoints/v1-5-pruned-emaonly.safetensors \
--model_tuned ~/sd/ComfyUI/models/checkpoints/spin-diffusion-v3-sd15.safetensors \
--dim 256 \
--save_precision fp16 \
--save_to ~/sd/ComfyUI/models/loras/spin-diffusion-v3-256.ckpt


./venv/bin/python networks/extract_lora_from_models.py \
--model_org ~/sd/ComfyUI/models/checkpoints/v1-5-pruned-emaonly.safetensors \
--model_tuned ~/sd/ComfyUI/models/checkpoints/spin-diffusion-v3-sd15.safetensors \
--dim 128 \
--save_precision fp16 \
--save_to ~/sd/ComfyUI/models/loras/spin-diffusion-v3-128.ckpt