Update app.py

This commit is contained in:
sigma
2026-01-24 04:56:12 +00:00
committed by system
parent 38a73d971e
commit 6f81898c09

8
app.py
View File

@@ -277,15 +277,15 @@ pipe = QwenImageEditPlusPipeline.from_pretrained(
# 2. DOWNLOAD & LOAD RAW WEIGHTS # 2. DOWNLOAD & LOAD RAW WEIGHTS
# ------------------------------------------------------------------------------ # ------------------------------------------------------------------------------
print("accessing v21 checkpoint...") print("accessing v23 checkpoint...")
v21_path = hf_hub_download( v23_path = hf_hub_download(
repo_id="Phr00t/Qwen-Image-Edit-Rapid-AIO", repo_id="Phr00t/Qwen-Image-Edit-Rapid-AIO",
filename="v21/Qwen-Rapid-AIO-NSFW-v21.safetensors", filename="v23/Qwen-Rapid-AIO-NSFW-v23.safetensors",
repo_type="model" repo_type="model"
) )
print(f"loading 28GB state dict into cpu memory...") print(f"loading 28GB state dict into cpu memory...")
state_dict = load_file(v21_path) state_dict = load_file(v23_path)
# 3. DYNAMIC COMPONENT MAPPING (NO ASSUMPTIONS) # 3. DYNAMIC COMPONENT MAPPING (NO ASSUMPTIONS)
# ------------------------------------------------------------------------------ # ------------------------------------------------------------------------------