echo "Creating python venv..."
python -m venv venv
source venv/bin/activate

echo "Installing torch & xformers..."
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118
pip install xformers==0.0.18
echo "Installing deps..."
cd ./sd-scripts

pip install --upgrade -r requirements.txt
pip install --upgrade lion-pytorch lycoris-lora dadaptation

echo "Install completed"