Update app.py

app.py

@@ -1,8 +1,7 @@
 import os
 import spaces
 import torch
-from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
-from safetensors.torch import load_file
+from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
 import gradio as gr
 import random
 import tqdm
@@ -18,17 +17,13 @@ def load_model():
         filename="AkashicPulse-v1.0-ft-ft.safetensors"
     )
     
-    # Initialize pipeline with base SDXL configuration
-    pipe = StableDiffusionXLPipeline.from_pretrained(
-        "stabilityai/stable-diffusion-xl-base-1.0",
+    # Initialize standard SD pipeline
+    pipe = StableDiffusionPipeline.from_single_file(
+        model_path,
         torch_dtype=torch.float16,
-        variant="fp16",
         use_safetensors=True
     )
     
-    # Load the custom model weights
-    state_dict = load_file(model_path)
-    pipe.unet.load_state_dict(state_dict)
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     return pipe