Update app.py

app.py

@@ -17,7 +17,7 @@ pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 2048
+MAX_IMAGE_SIZE = 1024
 
 @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, progress=gr.Progress(track_tqdm=True)):
@@ -91,7 +91,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=MAX_IMAGE_SIZE,
                     step=32,
-                    value=2048, #Replace with defaults that work for your model
+                    value=1024, #Replace with defaults that work for your model
                 )
                 
                 height = gr.Slider(
@@ -99,7 +99,7 @@ with gr.Blocks(css=css) as demo:
                     minimum=256,
                     maximum=MAX_IMAGE_SIZE,
                     step=32,
-                    value=2048, #Replace with defaults that work for your model
+                    value=1024, #Replace with defaults that work for your model
                 )
             
             with gr.Row():