Skip to content

Commit

Permalink
fix: openclip modelname
Browse files Browse the repository at this point in the history
  • Loading branch information
numb3r3 committed Jul 26, 2022
1 parent 88a1f99 commit add478c
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions server/clip_server/model/openclip_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down

0 comments on commit add478c

Please sign in to comment.