diffusers: fix set-sampler error following model switch

keturn · keturn · commit fdf2ed258a1d · 2022-11-11T13:17:36.000-08:00
diff --git a/ldm/generate.py b/ldm/generate.py
@@ -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,

Original file line number	Diff line number	Diff line change
`@@ -847,8 +847,8 @@ def set_model(self,model_name):`
`847`	`847`	`self.embedding_path, self.precision == 'float32' or self.precision == 'autocast'`
`848`	`848`	`)`
`849`	`849`
`850`		`- self._set_sampler()`
`851`	`850`	`self.model_name = model_name`
	`851`	`+ self._set_sampler() # requires self.model_name to be set first`
`852`	`852`	`return self.model`
`853`	`853`
`854`	`854`	`def correct_colors(self,`