Update app.py

This commit is contained in:
sigma
2026-01-19 01:08:14 +00:00
committed by system
parent 6480e58412
commit 8999693eee

16
app.py
View File

@@ -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,