Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 import numpy as np
 import modin.pandas as pd
 from PIL import Image
-from diffusers import DiffusionPipeline, StableDiffusion3Pipeline
+from diffusers import DiffusionPipeline, #StableDiffusion3Pipeline
 from huggingface_hub import hf_hub_download
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -58,7 +58,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
         
         torch.cuda.empty_cache()
         torch.cuda.max_memory_allocated(device=device)
-        sdxl = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16)#variant="fp16", use_safetensors=True)
+        sdxl = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
         sdxl.enable_xformers_memory_efficient_attention()
         sdxl = sdxl.to(device)   
         torch.cuda.empty_cache()