Update app.py
This commit is contained in:
16
app.py
16
app.py
@@ -245,12 +245,12 @@ pipe = QwenImageEditPlusPipeline.from_pretrained(
|
|||||||
torch_dtype=dtype
|
torch_dtype=dtype
|
||||||
).to(device)
|
).to(device)
|
||||||
|
|
||||||
print("loading lightning lora...")
|
# print("loading lightning lora...")
|
||||||
pipe.load_lora_weights(
|
# pipe.load_lora_weights(
|
||||||
"lightx2v/Qwen-Image-Edit-2511-Lightning",
|
# "lightx2v/Qwen-Image-Edit-2511-Lightning",
|
||||||
weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors"
|
# weight_name="Qwen-Image-Edit-2511-Lightning-4steps-V1.0-bf16.safetensors"
|
||||||
)
|
# )
|
||||||
pipe.fuse_lora()
|
# pipe.fuse_lora()
|
||||||
print("lightning lora fused.")
|
print("lightning lora fused.")
|
||||||
|
|
||||||
|
|
||||||
@@ -296,8 +296,8 @@ def infer(
|
|||||||
prompt,
|
prompt,
|
||||||
seed=42,
|
seed=42,
|
||||||
randomize_seed=False,
|
randomize_seed=False,
|
||||||
true_guidance_scale=1.0,
|
true_guidance_scale=4.0,
|
||||||
num_inference_steps=4,
|
num_inference_steps=40,
|
||||||
height=None,
|
height=None,
|
||||||
width=None,
|
width=None,
|
||||||
rewrite_prompt=True,
|
rewrite_prompt=True,
|
||||||
|
|||||||
Reference in New Issue
Block a user