Hugging Face's logo

Spaces:
LWZ19
/
ecodiff_flux_dev
Sleeping

App Files Community
LWZ19 commited on
Commit
183f9cb
·
verified ·
1 Parent(s): 55b55d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -80,7 +80,7 @@ for ratio in PRUNING_RATIOS:
80
  print("📥 Preloading LoRA checkpoint for 20% pruning ratio...")
81
  try:
82
  lora_repo_path = snapshot_download(
83
- repo_id="LWZ19/flux_dev_20_ckp",
84
  allow_patterns=[f"lora/checkpoint-{LORA_CHECKPOINT_STEP}/*"]
85
  )
86
  lora_weights = load_file(os.path.join(lora_repo_path, "lora", f"checkpoint-{LORA_CHECKPOINT_STEP}", LORA_WEIGHT_NAME_SAFE))
 
80
  print("📥 Preloading LoRA checkpoint for 20% pruning ratio...")
81
  try:
82
  lora_repo_path = snapshot_download(
83
+ repo_id="LWZ19/flux_dev_20_ckp_2",
84
  allow_patterns=[f"lora/checkpoint-{LORA_CHECKPOINT_STEP}/*"]
85
  )
86
  lora_weights = load_file(os.path.join(lora_repo_path, "lora", f"checkpoint-{LORA_CHECKPOINT_STEP}", LORA_WEIGHT_NAME_SAFE))