Hi I'm trying to use Shivam colab with a custom mode ( Realisticvision 1.3) from hugging face. It ha

Hi I'm trying to use Shivam colab with a custom mode ( Realisticvision 1.3) from hugging face. It has all the diffusers data. In the colab, I can run the inference without problem and generate images, but when I try to train it exits with errors. I commented out the "fp16 revison" flag and changed the VAE parameter to " --pretrained_vae_name_or_path="SG161222/Realistic_Vision_V1.3/vae" "

I got "subprocess.CalledProcessError: Command '['/usr/bin/python3', 'train_dreambooth.py', '--pretrained_model_name_or_path=SG161222/Realistic_Vision_V1.3', '--pretrained_vae_name_or_path=SG161222/Realistic_Vision_V1.3/vae', '--output_dir=/content/Rv2_weights/zwx']' returned non-zero exit status 1.
"
Do you know what the issue is? Please see the screenshots for more details. Thanks!
Screenshot_2023-02-17_at_7.05.06_AM.png
Screenshot_2023-02-17_at_7.07.48_AM.png
Was this page helpful?