Skip to content

Fix hard-coded diffusion model type in trainer#16

Open
jvwilliams23 wants to merge 1 commit intoBaratiLab:main_v1from
jvwilliams23:main_v1
Open

Fix hard-coded diffusion model type in trainer#16
jvwilliams23 wants to merge 1 commit intoBaratiLab:main_v1from
jvwilliams23:main_v1

Conversation

@jvwilliams23
Copy link

I was getting an error when trying to train with the model type "simple", due to hard-coded ConditionalDiffusion. I added a check to see which model to use.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant