Spaces:
Running
on
Zero
Running
on
Zero
qiaochanghao
commited on
Commit
·
90e2903
1
Parent(s):
a063e5b
change model
Browse files
app.py
CHANGED
|
@@ -153,7 +153,7 @@ dtype = torch.bfloat16
|
|
| 153 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 154 |
|
| 155 |
# Load the model pipeline
|
| 156 |
-
pipe = QwenImageEditPlusPipeline.from_pretrained("
|
| 157 |
|
| 158 |
# --- UI Constants and Helpers ---
|
| 159 |
MAX_SEED = np.iinfo(np.int32).max
|
|
|
|
| 153 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 154 |
|
| 155 |
# Load the model pipeline
|
| 156 |
+
pipe = QwenImageEditPlusPipeline.from_pretrained("FireRedTeam/FireRed-Image-Edit-1.0", torch_dtype=dtype).to(device)
|
| 157 |
|
| 158 |
# --- UI Constants and Helpers ---
|
| 159 |
MAX_SEED = np.iinfo(np.int32).max
|