From 8b90a87e92c98535d70a1c29a2ee1bf9d1f233b8 Mon Sep 17 00:00:00 2001 From: numb3r3 Date: Mon, 25 Jul 2022 17:27:01 +0800 Subject: [PATCH] fix: openclip modelname --- server/clip_server/model/openclip_model.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/server/clip_server/model/openclip_model.py b/server/clip_server/model/openclip_model.py index dcfc39da4..3d667ebcf 100644 --- a/server/clip_server/model/openclip_model.py +++ b/server/clip_server/model/openclip_model.py @@ -40,8 +40,6 @@ def get_model_name(name: str): model_name = name if model_name == 'ViT-L/14@336px': return 'ViT-L-14-336' - elif model_name.endswith('-quickgelu'): - return model_name[:-10] return model_name.replace('/', '-') def encode_text(self, input_ids: 'torch.Tensor', **kwargs):