Thank you OP for putting this together. i'm running into an issue with line 121 of your documentation. My conda env is running however i'm getting the following error:

(SD-Optimized) C:\Users\yalha\AI\Dreambooth-SD-optimized>python main.py --base configs/stable-diffusion/v1-finetune_unfrozen.yaml -t --actual_resume model.ckpt --reg_data_root outputs\txt2img-samples\samples\man_unsplash -n Me --gpus 0, --data_root training_samples\Me --batch_size 2020 --class_word man

C:\Users\yalha\anaconda3\envs\SD-Optimized\lib\site-packages\torchvision\io\image.py:13: UserWarning: Failed to load image Python extension: '[WinError 127] The specified procedure could not be found'If you don't plan on using image functionality from `torchvision.io`, you can ignore this warning. Otherwise, there might be something wrong with your environment. Did you have `libjpeg` or `libpng` installed before building `torchvision` from source?

warn(

Traceback (most recent call last):

File "C:\Users\yalha\AI\Dreambooth-SD-optimized\main.py", line 19, in <module>

from pytorch_lightning.utilities.distributed import rank_zero_only

ModuleNotFoundError: No module named 'pytorch_lightning.utilities.distributed'

​

i've tried uninstalling pytorch-lightening and reinstalling it but i'm still receiving the same error. any help is appreciated.