KingNish commited on
Commit
bf07966
·
verified ·
1 Parent(s): ed0f679

batch size 12

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -29,7 +29,7 @@ if torch.cuda.is_available():
29
  add_watermarker=False
30
  )
31
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
32
- pipe.load_lora_weights("ehristoforu/dalle-3-xl", weight_name="dalle-3-xl-lora-v1.safetensors", adapter_name="dalle")
33
  pipe.set_adapters("dalle")
34
 
35
  pipe.to("cuda")
@@ -184,6 +184,8 @@ with gr.Blocks(css=css) as demo:
184
  negative_prompt.submit,
185
  run_button.click,
186
  ],
 
 
187
  fn=generate,
188
  inputs=[
189
  prompt,
 
29
  add_watermarker=False
30
  )
31
  pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
32
+ pipe.load_lora_weights("ehristoforu/dalle-3-xl-v2", weight_name="dalle-3-xl-lora-v2.safetensors", adapter_name="dalle")
33
  pipe.set_adapters("dalle")
34
 
35
  pipe.to("cuda")
 
184
  negative_prompt.submit,
185
  run_button.click,
186
  ],
187
+ batch=True,
188
+ max_batch_size=12,
189
  fn=generate,
190
  inputs=[
191
  prompt,