From 88a8c30708a157a40d74201a73b3e3fe1294acd9 Mon Sep 17 00:00:00 2001
From: Jael Gu <mengjia.gu@zilliz.com>
Date: Fri, 13 Jan 2023 11:39:46 +0800
Subject: [PATCH] Deal with model & Model

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
---
 auto_transformers.py | 4 ----
 1 file changed, 4 deletions(-)

diff --git a/auto_transformers.py b/auto_transformers.py
index 02e0f2a..a6a47bb 100644
--- a/auto_transformers.py
+++ b/auto_transformers.py
@@ -25,7 +25,6 @@ from transformers import AutoTokenizer, AutoConfig, AutoModel
 
 from towhee.operator import NNOperator
 from towhee import register
-# from towhee.serve.triton.triton_client import TritonClient
 # from towhee.dc2 import accelerate
 
 import warnings
@@ -132,9 +131,6 @@ class AutoTransformers(NNOperator):
 
     @property
     def _model(self):
-        # if isinstance(self.model, TritonClient):
-        #     model = create_model(self.model_name, self.checkpoint_path, self.device)
-        # else:
         model = self.model.model
         return model