sentence-embedding
copied
Remove test message
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
@ -73,7 +73,6 @@ class AutoTransformers(NNOperator):
else:
self.device = 'cuda' if torch.cuda.is_available() else 'cpu'
self.model_name = self.map_model_names(model_name)
print(1111, self.model_name)
if tokenizer:
self.tokenizer = tokenizer