Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -52,13 +52,13 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, randomize_seed, seed, wi
|
|
52 |
# Generate image using the pipeline
|
53 |
image = pipe(
|
54 |
prompt=f"{prompt} {trigger_word}",
|
55 |
-
negative_prompt,
|
56 |
num_inference_steps=steps,
|
57 |
guidance_scale=cfg_scale,
|
58 |
width=width,
|
59 |
height=height,
|
60 |
generator=generator,
|
61 |
-
joint_attention_kwargs={"scale": lora_scale}
|
62 |
).images[0]
|
63 |
|
64 |
# Save the image to a file with a unique name in /tmp directory
|
|
|
52 |
# Generate image using the pipeline
|
53 |
image = pipe(
|
54 |
prompt=f"{prompt} {trigger_word}",
|
55 |
+
negative_prompt=negative_prompt,
|
56 |
num_inference_steps=steps,
|
57 |
guidance_scale=cfg_scale,
|
58 |
width=width,
|
59 |
height=height,
|
60 |
generator=generator,
|
61 |
+
joint_attention_kwargs={"scale": lora_scale},
|
62 |
).images[0]
|
63 |
|
64 |
# Save the image to a file with a unique name in /tmp directory
|