Browse Source
Change accelerate to comment since not yet supported in latest release
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
2 additions and
2 deletions
-
auto_transformers.py
|
|
@ -22,7 +22,7 @@ from transformers import AutoTokenizer, AutoModel, AutoModelForMaskedLM, AutoMod |
|
|
|
|
|
|
|
|
from towhee.operator import NNOperator |
|
|
from towhee.operator import NNOperator |
|
|
from towhee import register |
|
|
from towhee import register |
|
|
from towhee.dc2 import accelerate |
|
|
|
|
|
|
|
|
# from towhee.dc2 import accelerate |
|
|
|
|
|
|
|
|
import warnings |
|
|
import warnings |
|
|
import logging |
|
|
import logging |
|
|
@ -35,7 +35,7 @@ log = logging.getLogger('run_op') |
|
|
warnings.filterwarnings('ignore') |
|
|
warnings.filterwarnings('ignore') |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@accelerate |
|
|
|
|
|
|
|
|
# @accelerate |
|
|
class Model: |
|
|
class Model: |
|
|
def __init__(self, model_name, device, checkpoint_path): |
|
|
def __init__(self, model_name, device, checkpoint_path): |
|
|
try: |
|
|
try: |
|
|
|
