Training script

[OrangeSodahub]

Hi, I wonder if I could use this script to train MVDream, I tried directly load SD2.1 from huggingface and it failed.

[ashawkey]

Sorry that this code is only for inference, but you should be able to load it. Can you give detailed error message and environment?

[OrangeSodahub]

The errors should be easy to fix. Just some extra arguments exist in yours __init__, while they are not given in diffusers's from_pretrained method

[ashawkey]

What's the version of diffusers are you using? It should work on the latest version.