FireRed-Image-Edit-1.1-Fast

Paused

WarlordHermes commited on 27 days ago

Commit

69eef46

verified ·

1 Parent(s): 678f688

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -116,7 +116,7 @@ dtype = torch.bfloat16
 pipe = QwenImageEditPlusPipeline.from_pretrained(
     "FireRedTeam/FireRed-Image-Edit-1.1",
     transformer=QwenImageTransformer2DModel.from_pretrained(
-        "prithivMLmods/Qwen-Image-Edit-Rapid-AIO-V23",
         torch_dtype=dtype,
         device_map="cuda",
     ),

 pipe = QwenImageEditPlusPipeline.from_pretrained(
     "FireRedTeam/FireRed-Image-Edit-1.1",
     transformer=QwenImageTransformer2DModel.from_pretrained(
+        "prithivMLmods/Qwen-Image-Edit-Rapid-AIO-V4",
         torch_dtype=dtype,
         device_map="cuda",
     ),