Closed ExponentialML closed 2 years ago
Does that solve it?
Thanks, you led me in the right direction. I ended up retraining the model because it seems like I needed to use --learn_sigma True
when training. I also retrained on a cosine
schedule, and had to add in the cosine parts from the guided-diffusion repository since they were removed from this one.
After doing that and changing the configuration, everything worked. I'll post the configuration when I'm able to, then I'll close the issue.
Great, thanks!
Hello, thanks for the release. I've trained a diffusion model using the
guided-diffusion
repository, but I can't seem to load the model. It's trained on a resolution of 64x64. This is a trained diffusion model using theimproved-diffusion
repository, not a classifier.When setting the parameters and doing inference, I'm getting an error. I've tested both trained classifiers and diffusion models on
guided-diffusion
, as well asimproved-diffusion
, and both have worked without issue.With RePaint, the error is as follows:
My model was trained using these parameters:
With this custom yaml configuration (I'm not using a classifier here as I can't get the diffusion model to load):
Any help would be appreciated!