Browse Source
Update accelerate
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
main
1 changed files with
5 additions and
1 deletions
-
auto_transformers.py
|
@ -25,7 +25,11 @@ from transformers import AutoTokenizer, AutoConfig, AutoModel, AutoModelForMaske |
|
|
|
|
|
|
|
|
from towhee.operator import NNOperator |
|
|
from towhee.operator import NNOperator |
|
|
from towhee import register |
|
|
from towhee import register |
|
|
# from towhee.dc2 import accelerate |
|
|
|
|
|
|
|
|
try: |
|
|
|
|
|
from towhee import accelerate |
|
|
|
|
|
except: |
|
|
|
|
|
def accelerate(func): |
|
|
|
|
|
return func |
|
|
|
|
|
|
|
|
import warnings |
|
|
import warnings |
|
|
import logging |
|
|
import logging |
|
|