This commit is contained in:
Timothy J. Baek 2024-09-19 22:40:06 +02:00
parent 619dbbe9f5
commit 39c57c0e94

View File

@ -212,11 +212,9 @@ def update_reranking_model(
# and the following error is thrown: # 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 # /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" lock_file = "/root/.cache/torch_extensions/py311_cpu/segmented_maxsim_cpp/lock"
try: if os.path.exists(lock_file):
shutil.rmtree(torch_extensions) os.remove(lock_file)
except:
pass
self.ckpt = Checkpoint( self.ckpt = Checkpoint(
name, name,