Update app.py
Browse files
app.py
CHANGED
|
@@ -116,7 +116,7 @@ dtype = torch.bfloat16
|
|
| 116 |
pipe = QwenImageEditPlusPipeline.from_pretrained(
|
| 117 |
"FireRedTeam/FireRed-Image-Edit-1.1",
|
| 118 |
transformer=QwenImageTransformer2DModel.from_pretrained(
|
| 119 |
-
"prithivMLmods/Qwen-Image-Edit-Rapid-AIO-
|
| 120 |
torch_dtype=dtype,
|
| 121 |
device_map="cuda",
|
| 122 |
),
|
|
@@ -426,8 +426,8 @@ with gr.Blocks(css=css, theme=theme, title="🔥 FireRed Image Edit") as demo:
|
|
| 426 |
<a href="https://huggingface.co/FireRedTeam/FireRed-Image-Edit-1.1"
|
| 427 |
target="_blank">FireRed-Image-Edit-1.1</a>
|
| 428 |
&
|
| 429 |
-
<a href="https://huggingface.co/prithivMLmods/Qwen-Image-Edit-Rapid-AIO-
|
| 430 |
-
target="_blank">Rapid-AIO-
|
| 431 |
</p>
|
| 432 |
<div class="badges">
|
| 433 |
<span class="bdg">⚡ 4-Step Fast</span>
|
|
|
|
| 116 |
pipe = QwenImageEditPlusPipeline.from_pretrained(
|
| 117 |
"FireRedTeam/FireRed-Image-Edit-1.1",
|
| 118 |
transformer=QwenImageTransformer2DModel.from_pretrained(
|
| 119 |
+
"prithivMLmods/Qwen-Image-Edit-Rapid-AIO-V23",
|
| 120 |
torch_dtype=dtype,
|
| 121 |
device_map="cuda",
|
| 122 |
),
|
|
|
|
| 426 |
<a href="https://huggingface.co/FireRedTeam/FireRed-Image-Edit-1.1"
|
| 427 |
target="_blank">FireRed-Image-Edit-1.1</a>
|
| 428 |
&
|
| 429 |
+
<a href="https://huggingface.co/prithivMLmods/Qwen-Image-Edit-Rapid-AIO-V23"
|
| 430 |
+
target="_blank">Rapid-AIO-V23</a>
|
| 431 |
</p>
|
| 432 |
<div class="badges">
|
| 433 |
<span class="bdg">⚡ 4-Step Fast</span>
|