fix: sdxl-turbo cpu dtype

luxonis · Feb 25, 2025 · 832df05 · 832df05
1 parent 07dc932
commit 832df05
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/datadreamer/image_generation/sdxl_turbo_image_generator.py b/datadreamer/image_generation/sdxl_turbo_image_generator.py
@@ -41,8 +41,6 @@ def _init_gen_model(self) -> AutoPipelineForText2Image:
         if self.device == "cpu":
             base = AutoPipelineForText2Image.from_pretrained(
                 "stabilityai/sdxl-turbo",
-                # variant="fp16",
-                torch_dtype=torch.float32,
                 use_safetensors=True,
             )
             base.to("cpu")