WithAnyone commited on
Commit
5688605
·
verified ·
1 Parent(s): 8b424c5

Update withanyone/flux/util.py

Browse files
Files changed (1) hide show
  1. withanyone/flux/util.py +2 -0
withanyone/flux/util.py CHANGED
@@ -227,6 +227,8 @@ def load_flow_model_no_lora(
227
  ):
228
  ckpt_path = hf_hub_download(configs[name].repo_id, configs[name].repo_flow.replace("sft", "safetensors"))
229
  print("Downloading checkpoint from HF:", ckpt_path)
 
 
230
  else:
231
  ckpt_path = os.path.join(path, "flux1-dev.safetensors") if path is not None else None
232
 
 
227
  ):
228
  ckpt_path = hf_hub_download(configs[name].repo_id, configs[name].repo_flow.replace("sft", "safetensors"))
229
  print("Downloading checkpoint from HF:", ckpt_path)
230
+ # remove xet to save space
231
+ os.remove("/home/user/.cache/huggingface/xet/")
232
  else:
233
  ckpt_path = os.path.join(path, "flux1-dev.safetensors") if path is not None else None
234