emd

Mikubill · Nov 1, 2023 · cdc4af5 · cdc4af5
1 parent 304de03
commit cdc4af5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/annotator/clipvision/__init__.py b/annotator/clipvision/__init__.py
@@ -80,6 +80,9 @@
 clip_vision_h_uc = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'clip_vision_h_uc.data')
 clip_vision_h_uc = torch.load(clip_vision_h_uc)['uc']
 
+clip_vision_vith_uc = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'clip_vision_vith_uc.data')
+clip_vision_vith_uc = torch.load(clip_vision_vith_uc)['uc']
+
 
 class ClipVisionDetector:
     def __init__(self, config):

diff --git a/annotator/clipvision/clip_vision_vith_uc.data b/annotator/clipvision/clip_vision_vith_uc.data