Skip to content

Commit

Permalink
diffusers: fix set-sampler error following model switch
Browse files Browse the repository at this point in the history
  • Loading branch information
keturn committed Nov 11, 2022
1 parent 7904d0c commit fdf2ed2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ldm/generate.py
Original file line number Diff line number Diff line change
Expand Up @@ -847,8 +847,8 @@ def set_model(self,model_name):
self.embedding_path, self.precision == 'float32' or self.precision == 'autocast'
)

self._set_sampler()
self.model_name = model_name
self._set_sampler() # requires self.model_name to be set first
return self.model

def correct_colors(self,
Expand Down

0 comments on commit fdf2ed2

Please sign in to comment.