diff --git a/ldm/generate.py b/ldm/generate.py index 678c473a32d..8fb0ca91473 100644 --- 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,