logo
Browse Source

Update accelerate

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
Jael Gu 2 years ago
parent
commit
488c4cc394
  1. 6
      auto_transformers.py

6
auto_transformers.py

@ -23,7 +23,11 @@ from transformers import AutoTokenizer, AutoModel, AutoModelForMaskedLM, AutoMod
from towhee.operator import NNOperator
from towhee import register
# from towhee.dc2 import accelerate
try:
from towhee import accelerate
except:
def accelerate(func):
return func
import warnings
import logging

Loading…
Cancel
Save