Update withanyone/flux/util.py

withanyone/flux/util.py

@@ -228,7 +228,8 @@ def load_flow_model_no_lora(
         ckpt_path = hf_hub_download(configs[name].repo_id, configs[name].repo_flow.replace("sft", "safetensors"))
         print("Downloading checkpoint from HF:", ckpt_path)
         # remove xet to save space
-        os.remove("/home/user/.cache/huggingface/xet/")
+        import shutil
+        shutil.rmtree("/home/user/.cache/huggingface/xet/")
     else:
         ckpt_path = os.path.join(path, "flux1-dev.safetensors") if path is not None else None