search

robvanvolt / DALLE-models

Here is a collection of checkpoints for DALLE-pytorch models, from where you can keep on training or start generating images.
MIT License
146 stars 13 forks source link

RuntimeError: Error(s) in loading state_dict for DALLE #15

Open jrryzh opened 2 years ago

jrryzh commented 2 years ago

During trying to load the dalle pre-trained weights, I meet a loading error, and it seems that the current dalle model has different structures and different weight keys. I'm guessing I am using the newer version of dalle-pytorch(dalle-pytorch-1.5.2). Can you specify the supported version of dalle-pytorch? Thank u so much! image

GaiAshkenazyVia commented 2 years ago

Same problem - I tried few versions none of them worked for me and I ran into the same error

basisvectors commented 2 years ago

Still have the same problem, all 6 models give the same runtime error as yours. The notebook is unusable.