signsur4739379373 commited on
Commit
6f81898
·
verified ·
1 Parent(s): 38a73d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -277,15 +277,15 @@ pipe = QwenImageEditPlusPipeline.from_pretrained(
277
 
278
  # 2. DOWNLOAD & LOAD RAW WEIGHTS
279
  # ------------------------------------------------------------------------------
280
- print("accessing v21 checkpoint...")
281
- v21_path = hf_hub_download(
282
  repo_id="Phr00t/Qwen-Image-Edit-Rapid-AIO",
283
- filename="v21/Qwen-Rapid-AIO-NSFW-v21.safetensors",
284
  repo_type="model"
285
  )
286
 
287
  print(f"loading 28GB state dict into cpu memory...")
288
- state_dict = load_file(v21_path)
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
  # ------------------------------------------------------------------------------