MihaiHuggingFace commited on
Commit
bc7f486
·
verified ·
1 Parent(s): b1301a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,7 +12,7 @@ torch.backends.cuda.matmul.allow_tf32 = True
12
 
13
  # Constants
14
  MAX_SEED = np.iinfo(np.int32).max
15
- MAX_IMAGE_SIZE = 2048
16
  DEFAULT_WIDTH = 1024
17
  DEFAULT_HEIGHT = 1024
18
  DEFAULT_INFERENCE_STEPS = 1
@@ -91,8 +91,8 @@ with gr.Blocks() as demo:
91
  seed = gr.Number(label="Seed", value=42)
92
  randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
93
  with gr.Row():
94
- width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_WIDTH)
95
- height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_HEIGHT)
96
  num_inference_steps = gr.Slider(label="Inference Steps", minimum=1, maximum=4, step=1, value=DEFAULT_INFERENCE_STEPS)
97
 
98
  with gr.Row():
 
12
 
13
  # Constants
14
  MAX_SEED = np.iinfo(np.int32).max
15
+ MAX_IMAGE_SIZE = 4096
16
  DEFAULT_WIDTH = 1024
17
  DEFAULT_HEIGHT = 1024
18
  DEFAULT_INFERENCE_STEPS = 1
 
91
  seed = gr.Number(label="Seed", value=42)
92
  randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
93
  with gr.Row():
94
+ width = gr.Slider(label="Width", minimum=128, maximum=MAX_IMAGE_SIZE, step=16, value=DEFAULT_WIDTH)
95
+ height = gr.Slider(label="Height", minimum=128, maximum=MAX_IMAGE_SIZE, step=16, value=DEFAULT_HEIGHT)
96
  num_inference_steps = gr.Slider(label="Inference Steps", minimum=1, maximum=4, step=1, value=DEFAULT_INFERENCE_STEPS)
97
 
98
  with gr.Row():