Typo

MMqd · Jul 23, 2023 · 008bec9 · 008bec9
1 parent a9a4e29
commit 008bec9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/deepfloydif.py b/scripts/deepfloydif.py
@@ -66,7 +66,7 @@ def txt2img(self, p, generation_parameters, b):
         generation_parameters.pop("negative_prompt", None)
         result_images = self.pipe(**generation_parameters, num_images_per_prompt=p.batch_size).images
         if self.stageII_model != "None":
-            self.pipe = self.load_pipeline("pipe", IFSuperResolutionPipeline, f"DeepFloyd/IF-I-{self.stageII_model}-v1.0", {"safety_checker": None, "watermarker": None})
+            self.pipe = self.load_pipeline("pipe", IFSuperResolutionPipeline, f"DeepFloyd/IF-II-{self.stageII_model}-v1.0", {"safety_checker": None, "watermarker": None})
         generation_parameters["width"] = generation_parameters["width"]*4
         generation_parameters["height"] = generation_parameters["height"]*4
         result_images = self.pipe(**generation_parameters, num_images_per_prompt=p.batch_size).images