Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -277,15 +277,15 @@ pipe = QwenImageEditPlusPipeline.from_pretrained(
|
|
| 277 |
|
| 278 |
# 2. DOWNLOAD & LOAD RAW WEIGHTS
|
| 279 |
# ------------------------------------------------------------------------------
|
| 280 |
-
print("accessing
|
| 281 |
-
|
| 282 |
repo_id="Phr00t/Qwen-Image-Edit-Rapid-AIO",
|
| 283 |
-
filename="
|
| 284 |
repo_type="model"
|
| 285 |
)
|
| 286 |
|
| 287 |
print(f"loading 28GB state dict into cpu memory...")
|
| 288 |
-
state_dict = load_file(
|
| 289 |
|
| 290 |
# 3. DYNAMIC COMPONENT MAPPING (NO ASSUMPTIONS)
|
| 291 |
# ------------------------------------------------------------------------------
|
|
|
|
| 277 |
|
| 278 |
# 2. DOWNLOAD & LOAD RAW WEIGHTS
|
| 279 |
# ------------------------------------------------------------------------------
|
| 280 |
+
print("accessing v23 checkpoint...")
|
| 281 |
+
v23_path = hf_hub_download(
|
| 282 |
repo_id="Phr00t/Qwen-Image-Edit-Rapid-AIO",
|
| 283 |
+
filename="v23/Qwen-Rapid-AIO-NSFW-v23.safetensors",
|
| 284 |
repo_type="model"
|
| 285 |
)
|
| 286 |
|
| 287 |
print(f"loading 28GB state dict into cpu memory...")
|
| 288 |
+
state_dict = load_file(v23_path)
|
| 289 |
|
| 290 |
# 3. DYNAMIC COMPONENT MAPPING (NO ASSUMPTIONS)
|
| 291 |
# ------------------------------------------------------------------------------
|