chore: remove clip_hg

.github/workflows/cd.yml

@@ -42,7 +42,7 @@ jobs:
           python -m pip install wheel
           pip install --no-cache-dir "client/[test]"
           pip install --no-cache-dir "server/[onnx]"
-          pip install --no-cache-dir "server/[huggingface]"
+          pip install --no-cache-dir "server/[mclip]"
       - name: Test
         id: test
         run: |

.github/workflows/ci.yml

@@ -114,7 +114,7 @@ jobs:
           python -m pip install wheel pytest pytest-cov
           pip install --no-cache-dir "client/[test]"
           pip install --no-cache-dir "server/[onnx]"
-          pip install --no-cache-dir "server/[huggingface]"
+          pip install --no-cache-dir "server/[mclip]"
       - name: Test
         id: test
         run: |

server/setup.py

@@ -58,7 +58,7 @@
         ]
         + (['onnxruntime-gpu>=1.8.0'] if sys.platform != 'darwin' else []),
         'tensorrt': ['nvidia-tensorrt'],
-        'huggingface': ['transformers>=4.16.2'],
+        'mclip': ['transformers>=4.16.2'],
     },
     classifiers=[
         'Development Status :: 5 - Production/Stable',

tests/conftest.py

@@ -16,15 +16,13 @@ def random_port():
     return random_port
 
 
-@pytest.fixture(scope='session', params=['onnx', 'torch', 'hg', 'onnx_custom'])
+@pytest.fixture(scope='session', params=['onnx', 'torch', 'onnx_custom'])
 def make_flow(port_generator, request):
     if request.param != 'onnx_custom':
         if request.param == 'onnx':
             from clip_server.executors.clip_onnx import CLIPEncoder
-        elif request.param == 'torch':
-            from clip_server.executors.clip_torch import CLIPEncoder
         else:
-            from clip_server.executors.clip_hg import CLIPEncoder
+            from clip_server.executors.clip_torch import CLIPEncoder
 
         f = Flow(port=port_generator()).add(name=request.param, uses=CLIPEncoder)
     else:
@@ -56,25 +54,3 @@ def make_trt_flow(port_generator, request):
     f = Flow(port=port_generator()).add(name=request.param, uses=CLIPEncoder)
     with f:
         yield f
-
-
-@pytest.fixture(scope='session', params=['hg'])
-def make_hg_flow(port_generator, request):
-    from clip_server.executors.clip_hg import CLIPEncoder
-
-    f = Flow(port=port_generator()).add(name=request.param, uses=CLIPEncoder)
-    with f:
-        yield f
-
-
-@pytest.fixture(scope='session', params=['hg'])
-def make_hg_flow_no_default(port_generator, request):
-    from clip_server.executors.clip_hg import CLIPEncoder
-
-    f = Flow(port=port_generator()).add(
-        name=request.param,
-        uses=CLIPEncoder,
-        uses_with={'preprocessing': False},
-    )
-    with f:
-        yield f