Upload pipeline.py

pipeline.py

@@ -342,6 +342,7 @@ class SuperDiffSDXLPipeline(DiffusionPipeline, ConfigMixin):
         self.num_inference_steps = num_inference_steps
         self.guidance_scale = guidance_scale
         self.seed = seed
+        self.dtype = torch.float16
         if self.seed is None:
             self.seed = random.randint(0, 2**32 - 1)
 
@@ -354,6 +355,7 @@ class SuperDiffSDXLPipeline(DiffusionPipeline, ConfigMixin):
             generator=self.generator,
             dtype=self.dtype,
             device=self.device,
+            dtype=self.dtype,
         )
         prompt_embeds, added_cond_kwargs = self.prepare_prompt_input(
             prompt_1, prompt_2, batch_size, height, width