From 8752bfc1b09e7d7990ce024933dc070f36d8498e Mon Sep 17 00:00:00 2001
From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com>
Date: Tue, 20 Aug 2024 21:53:02 -0700
Subject: [PATCH] revise space

---
 modules_forge/forge_space.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/modules_forge/forge_space.py b/modules_forge/forge_space.py
index 3e34cb4d..afbce069 100644
--- a/modules_forge/forge_space.py
+++ b/modules_forge/forge_space.py
@@ -118,7 +118,10 @@ class ForgeSpace:
         return self.refresh_gradio()
 
     def uninstall(self):
-        shutil.rmtree(self.hf_path)
+        def on_rm_error(func, path, exc_info):
+            print(f"Error deleting {path}. Error: {exc_info[1]}")
+        
+        shutil.rmtree(self.hf_path, onerror=on_rm_error)
         print(f'Deleted: {self.hf_path}')
         return self.refresh_gradio()