Remove safetensors variable and changing default resolution.

app.py

@@ -14,7 +14,6 @@ pipe = StableDiffusionXLPipeline.from_pretrained(
     "kayfahaarukku/UrangDiffusion-1.0", 
     torch_dtype=torch.float16, 
     custom_pipeline="lpw_stable_diffusion_xl",
-    use_safetensors=True, 
 )
 pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
 
@@ -48,7 +47,6 @@ def generate_image(prompt, negative_prompt, use_defaults, resolution, guidance_s
     ).images[0]
 
     torch.cuda.empty_cache()
-    pipe.to('cpu')  # Move the model back to CPU after generation
 
     return image, seed
 
@@ -75,7 +73,7 @@ with gr.Blocks(title="UrangDiffusion 1.0 Demo", theme="NoCrypt/[email protected]") as d
                     "1344x768", "768x1344", "1536x640", "640x1536"
                 ],
                 label="Resolution",
-                value="896x1152"
+                value="832x1216"
             )
             guidance_scale_input = gr.Slider(minimum=1, maximum=20, step=0.5, label="Guidance Scale", value=7)
             num_inference_steps_input = gr.Slider(minimum=1, maximum=100, step=1, label="Number of Inference Steps", value=28)