Our methods code in mainly in .\MIX-OF-SHOW\custom_optimizers.py

1. Install requirements
pip install -r requirements.txt

2. Download pretrained model
mkdir experiments
mkdir experiments/pretrained_models
cd experiments/pretrained_models
git-lfs clone https://huggingface.co/windwhinny/chilloutmix.git

3. Create model directory
cd ../..    
mkdir experiments/8101_EDLoRA_potter_Cmix_B4_Repeat500
mkdir experiments/8101_EDLoRA_potter_Cmix_B4_Repeat500/models

4. Train (specify learning rates in -opt files)
accelerate launch train_edlora.py -opt options/train/EDLoRA/real/8101_EDLoRA_potter_Cmix_B4_Repeat500.yml --optimizer lift_adam

5. Create image directory
mkdir results
mkdir results/8101_EDLoRA_potter_Cmix_B4_Repeat500

6. Test
python test_edlora.py -opt options/test/EDLoRA/human/8101_EDLoRA_potter_Cmix_B4_Repeat500.yml

Figures will be saved to results/8101_EDLoRA_potter_Cmix_B4_Repeat500/visualization.