prithivMLmods commited on
Commit
c14ebf1
·
verified ·
1 Parent(s): 2ee3396

update app

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -101,7 +101,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
101
  pipe = QwenImageEditPlusPipeline.from_pretrained(
102
  "Qwen/Qwen-Image-Edit-2509",
103
  transformer=QwenImageTransformer2DModel.from_pretrained(
104
- "linoyts/Qwen-Image-Edit-Rapid-AIO",
105
  subfolder='transformer',
106
  torch_dtype=dtype,
107
  device_map='cuda'
 
101
  pipe = QwenImageEditPlusPipeline.from_pretrained(
102
  "Qwen/Qwen-Image-Edit-2509",
103
  transformer=QwenImageTransformer2DModel.from_pretrained(
104
+ "linoyts/Qwen-Image-Edit-Rapid-AIO", # [transformer weights extracted from: Phr00t/Qwen-Image-Edit-Rapid-AIO]
105
  subfolder='transformer',
106
  torch_dtype=dtype,
107
  device_map='cuda'