conda create -n dit4sr python=3.9 -y
conda activate dit4sr

pip install torch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 --index-url https://download.pytorch.org/whl/cu121
pip install -r requirements.txt
pip install numpy==1.26.3 --no-deps
pip install pyiqa==0.1.14.1 --no-deps 
cd detectron2 
pip install -e . --no-deps --no-build-isolation --config-settings editable_mode=compat
cd ../testr 
pip install -e . --no-deps --no-build-isolation --config-settings editable_mode=compat
pip install cloudpickle --no-deps

# First login enter your HF access token to download SD3 weights
huggingface-cli login   

# Then run the bash file
bash download_bash/download_sd3.sh 

bash download_bash/download_dit4sr.sh 

bash download_bash/download_testr.sh 

# stage1 training 
bash run_scripts/train/JIHYE_train_stage1_dit4sr.sh

# stage2 training 
bash run_scripts/train/JIHYE_train_stage2_testr.sh

# stage3 training 
bash run_scripts/train/JIHYE_train_stage3_dit4sr_testr.sh