logo
Browse Source

Fix load trainer issue

Signed-off-by: Kaiyuan Hu <kaiyuan.hu@zilliz.com>
main
Kaiyuan Hu 2 years ago
parent
commit
e5905a8d66
  1. 5
      isc.py

5
isc.py

@ -14,6 +14,7 @@
import logging
import os
import warnings
from typing import Union, List
from pathlib import Path
@ -22,6 +23,7 @@ from towhee.operator.base import NNOperator, OperatorFlag
from towhee.types.arg import arg, to_image_color
from towhee import register
from towhee.models import isc
import sys
# from towhee.dc2 import accelerate
import torch
@ -30,10 +32,9 @@ from torchvision import transforms
from PIL import Image as PILImage
import timm
import warnings
warnings.filterwarnings('ignore')
log = logging.getLogger('isc_op')
_ = sys.modules[__name__]
# @accelerate

Loading…
Cancel
Save