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