Update main.py

main.py

@@ -91,14 +91,19 @@ if not SSD_1B:
         variant=variant,
         cache_dir=cache_path,
     )
-    pipe = DiffusionPipeline.from_pretrained(
-        # "stabilityai/stable-diffusion-xl-base-1.0",
-        "stabilityai/sdxl-turbo",
-        unet=unet,
-        torch_dtype=torch_dtype,
-        variant=variant,
-        cache_dir=cache_path,
-    )
+
+    # "stabilityai/stable-diffusion-xl-base-1.0",
+    model_id"stabilityai/sdxl-turbo",
+    #pipe = DiffusionPipeline.from_pretrained(
+    #    model_id=model_id,
+    #    unet=unet,
+    #    torch_dtype=torch_dtype,
+    #    variant=variant,
+    #    cache_dir=cache_path,
+   # )
+    from diffusers import StableDiffusionPipeline
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+
 
     pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
     if torch.cuda.is_available():