Hugging Face's logo

stabilityai
/
sd-vae-ft-mse-original

Text-to-Image
stable-diffusion
stable-diffusion-diffusers
Model card Files Community
14

Fine Tuning the model

#8
by zee-1 - opened
Discussion
zee-1

I am trying to find tune the model with hugging face Trainer. And I have a custom dataset of 4000 images. I am getting some error like. tensor with 99xxxx elements cannot be converted to scalar. Is this the problem with my dataset? Or I have to change any layer of the model.

momentum99

I am trying to find tune the model with hugging face Trainer. And I have a custom dataset of 4000 images. I am getting some error like. tensor with 99xxxx elements cannot be converted to scalar. Is this the problem with my dataset? Or I have to change any layer of the model.

Are you in finetuning VAE, can you give some tutorials?

Sign up or log in to comment