#!/bin/bash
cd /workspace/*
mkdir -p tmp
rm -rf README.md
rm -rf s
rm -rf ss.sh

apt-get update
sudo apt install -y python3.10
sudo apt-get install -y python3.10-distutils
sudo apt-get install git-lfs
sudo apt install -y libtcmalloc-minimal4
apt-get -y install rsync
apt-get -y install zip

curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10
python3.10 -m pip install --upgrade pip
python3.10 -m pip install setuptools
python3.10 -m pip install requests
python3.10 -m pip install wget
python3.10 -m pip install gdown
python3.10 -m pip install pcloud
python3.10 -m pip install natsort
python3.10 -m pip install httpcore
python3.10 -m pip install pyyaml
python3.10 -m pip install numba

curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | sudo bash
git config --global user.name "anonymous"
git config --global user.email my@gmail.com
git lfs install --force
sudo apt update && sudo apt install -y libgl1-mesa-glx
sudo apt autoremove && sudo apt clean

cd /workspace/*/CU
curl -L -O https://huggingface.co/zuv0/test/resolve/main/CU.tar
tar -xf CU.tar
rm -rf CU.tar
\cp -f e "../e"
\cp -f run.sh "../run.sh"
\cp -f out.sh "../out.sh"
\cp -f gc.sh "../gc.sh"
\cp -f forcePull.sh "../forcePull.sh"
\cp -f myRun.ipynb "../myRun.ipynb"
curl -L -O https://huggingface.co/zuv0/SDXLF17/resolve/main/downs.sh
#git gc
if [ ! -d "models" ]; then
	git clone https://huggingface.co/zu01ze/modelsXL models
	cd models
	#mkdir -p loras
	ln -vs ./Lora ./loras
    #mkdir -p vae
    ln -vs ./VAE ./vae
	#mkdir -p checkpoints
	ln -vs ./Stable-diffusion ./checkpoints
fi

cd /workspace/*
sh ./CU/downs.sh $1 $2

#pip uninstall -y torch
#pip uninstall -y torchaudio
#pip uninstall -y torchsde
#pip uninstall -y torchvision

#pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu124

pip install -r ./CU/requirements.txt

#apt-get install -y libgl1-mesa-glx
#python3.10 ./SD/launch.py --reinstall-torch --no-download-sd-model --exit
sh forcePull.sh
sh gc.sh 1
sudo apt clean