fix: trt file path

jina-ai · Aug 3, 2022 · d8162d2 · d8162d2
1 parent 2c8629d
commit d8162d2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/server/clip_server/model/clip_trt.py b/server/clip_server/model/clip_trt.py
@@ -55,7 +55,9 @@ def __init__(
         super().__init__(name)
 
         if name in _MODELS:
-            cache_dir = os.path.expanduser(f'~/.cache/clip/{name.replace("/", "-")}')
+            cache_dir = os.path.expanduser(
+                f'~/.cache/clip/{name.replace("/", "-").replace("::", "-")}'
+            )
 
             self._textual_path = os.path.join(
                 cache_dir,