Update app.py

app.py

@@ -2,31 +2,49 @@ import os
 import spaces
 import torch
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
+from safetensors.torch import load_file
 import gradio as gr
 import random
 import tqdm
+from huggingface_hub import hf_hub_download
 
 # Enable TQDM progress tracking
 tqdm.monitor_interval = 0
 
-# Load the diffusion pipeline
-pipe = StableDiffusionXLPipeline.from_pretrained(
-    "kayfahaarukku/irAsu-1.0", 
-    torch_dtype=torch.float16, 
-    custom_pipeline="lpw_stable_diffusion_xl",
-)
-pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
+# Load the model from safetensors file
+def load_model():
+    model_path = hf_hub_download(
+        repo_id="kayfahaarukku/AkashicPulse-v1.0",
+        filename="AkashicPulse-v1.0-ft-ft.safetensors"
+    )
+    
+    # Initialize pipeline with base SDXL configuration
+    pipe = StableDiffusionXLPipeline.from_pretrained(
+        "stabilityai/stable-diffusion-xl-base-1.0",
+        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
+
+# Load the pipeline
+pipe = load_model()
 
 # Function to generate an image
-@spaces.GPU  # Adjust the duration as needed
+@spaces.GPU
 def generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_scale, num_inference_steps, seed, randomize_seed, progress=gr.Progress()):
-    pipe.to('cuda')  # Move the model to GPU when the function is called
+    pipe.to('cuda')
     
     if randomize_seed:
         seed = random.randint(0, 99999999)
     if use_defaults:
-        prompt = f"{prompt}, best quality, amazing quality, very aesthetic"
-        negative_prompt = f"nsfw, lowres, (bad), text, error, fewer, extra, missing, worst quality, jpeg artifacts, low quality, watermark, unfinished, displeasing, oldest, early, chromatic aberration, signature, extra digits, artistic error, username, scan, [abstract], {negative_prompt}"
+        prompt = f"{prompt}, masterpiece, best quality"
+        negative_prompt = f"lowres, bad anatomy, bad hands, text, error, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality, signature, watermark, username, blurry, {negative_prompt}"
     generator = torch.manual_seed(seed)
     
     def callback(step, timestep, latents):
@@ -58,13 +76,13 @@ def interface_fn(prompt, negative_prompt, use_defaults, resolution, guidance_sca
     return image, seed, gr.update(value=metadata_text)
 
 def reset_inputs():
-    return gr.update(value=''), gr.update(value=''), gr.update(value=True), gr.update(value='832x1216'), gr.update(value=4), gr.update(value=28), gr.update(value=0), gr.update(value=True), gr.update(value='')
+    return gr.update(value=''), gr.update(value=''), gr.update(value=True), gr.update(value='832x1216'), gr.update(value=7), gr.update(value=28), gr.update(value=0), gr.update(value=True), gr.update(value='')
 
-with gr.Blocks(title="irAsu 1.0 Demo", theme="NoCrypt/[email protected]") as demo:
+with gr.Blocks(title="AkashicPulse v1.0 Demo", theme="NoCrypt/[email protected]") as demo:
     gr.HTML(
-        "<h1>irAsu 1.0 Demo</h1>"
-        "This demo is intended to showcase what the model is capable of and is not intended to be the main generation platform. Results produced with Diffusers are not the best, and it's highly recommended for you to get the model running inside Stable Diffusion WebUI or ComfyUI."
-        )
+        "<h1>AkashicPulse v1.0 Demo</h1>"
+        "This demo showcases the AkashicPulse v1.0 model capabilities. For best results, it's recommended to run the model in Stable Diffusion WebUI or ComfyUI with MaHiRo CFG enabled."
+    )
     with gr.Row():
         with gr.Column():
             prompt_input = gr.Textbox(lines=2, placeholder="Enter prompt here", label="Prompt")
@@ -78,8 +96,8 @@ with gr.Blocks(title="irAsu 1.0 Demo", theme="NoCrypt/[email protected]") as demo:
                 label="Resolution",
                 value="832x1216"
             )
-            guidance_scale_input = gr.Slider(minimum=1, maximum=20, step=0.5, label="Guidance Scale", value=4)
-            num_inference_steps_input = gr.Slider(minimum=1, maximum=100, step=1, label="Number of Inference Steps", value=28)
+            guidance_scale_input = gr.Slider(minimum=4, maximum=10, step=0.5, label="Guidance Scale (CFG)", value=7)
+            num_inference_steps_input = gr.Slider(minimum=20, maximum=30, step=1, label="Number of Steps", value=28)
             seed_input = gr.Slider(minimum=0, maximum=999999999, step=1, label="Seed", value=0, interactive=True)
             randomize_seed_input = gr.Checkbox(label="Randomize Seed", value=True)
             generate_button = gr.Button("Generate")
@@ -97,14 +115,15 @@ with gr.Blocks(title="irAsu 1.0 Demo", theme="NoCrypt/[email protected]") as demo:
             gr.Markdown(
                 """
                 ### Recommended prompt formatting:
-                `1girl/1boy, character name, from what series, everything else in any order, best quality, amazing quality, very aesthetic,`
+                `1girl/1boy, character name, series, by artist name, the rest of the prompt, masterpiece, best quality`
 
-                **PS:** `best quality, amazing quality, very aesthetic,` is automatically added when "Use Default Quality Tags and Negative Prompt" is enabled
+                **PS:** `masterpiece, best quality` is automatically added when "Use Default Quality Tags and Negative Prompt" is enabled
 
-                ### Recommended settings:
-                - Steps: 25-30
-                - CFG: 3.5-5
-                - Sweet spot: 28 steps, 4 CFG
+                ### Current settings (recommended):
+                - Sampler: Euler a (fixed)
+                - Steps: 20-30 (sweet spot: 28)
+                - CFG: 4-10 (sweet spot: 7)
+                - Optional: Enable MaHiRo CFG in reForge or ComfyUI
                 """
             )