2511

pipeline_sdxs.py

@@ -24,7 +24,7 @@ class SdxsPipeline(DiffusionPipeline):
 
     def encode_prompt(self, prompt=None, negative_prompt=None, device=None, dtype=None):
         device = device or self.device
-        dtype = dtype or torch.float16  # Явно указываем float16
+        dtype = dtype or next(self.unet.parameters()).dtype
     
         # Преобразуем в списки
         if isinstance(prompt, str):
@@ -92,7 +92,9 @@ class SdxsPipeline(DiffusionPipeline):
         generator=None,
     ):
         device = self.device
-        dtype = torch.float16  # Явно указываем float16
+        dtype = next(self.unet.parameters()).dtype
+        
+        self.scheduler.set_timesteps(num_inference_steps, device=device)
     
         # Разделяем эмбеддинги на условные и безусловные
         if guidance_scale > 1: