Timothy J. Baek 7 tháng trước cách đây
mục cha
commit
39c57c0e94
1 tập tin đã thay đổi với 3 bổ sung5 xóa
  1. 3 5
      backend/open_webui/apps/rag/main.py

+ 3 - 5
backend/open_webui/apps/rag/main.py

@@ -212,11 +212,9 @@ def update_reranking_model(
                         # and the following error is thrown:
                         # /root/.cache/torch_extensions/py311_cpu/segmented_maxsim_cpp/segmented_maxsim_cpp.so: cannot open shared object file: No such file or directory
 
-                        torch_extensions = "/root/.cache/torch_extensions/py311_cpu"
-                        try:
-                            shutil.rmtree(torch_extensions)
-                        except:
-                            pass
+                        lock_file = "/root/.cache/torch_extensions/py311_cpu/segmented_maxsim_cpp/lock"
+                        if os.path.exists(lock_file):
+                            os.remove(lock_file)
 
                     self.ckpt = Checkpoint(
                         name,