Spaces:
Running
on
Zero
Running
on
Zero
batch size 12
Browse files
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-
|
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,
|