diff --git a/modelscope/__init__.py b/modelscope/__init__.py index 41b3e14c..b7712b3b 100644 --- a/modelscope/__init__.py +++ b/modelscope/__init__.py @@ -32,17 +32,17 @@ if TYPE_CHECKING: build_dataset_from_file) from .utils.constant import Tasks if is_transformers_available(): - from .utils.hf_util import AutoConfig, GPTQConfig, AwqConfig, BitsAndBytesConfig from .utils.hf_util import ( - AutoModel, AutoModelForCausalLM, AutoModelForSeq2SeqLM, + AutoModel, AutoProcessor, AutoFeatureExtractor, GenerationConfig, + AutoConfig, GPTQConfig, AwqConfig, BitsAndBytesConfig, + AutoModelForCausalLM, AutoModelForSeq2SeqLM, AutoModelForSequenceClassification, AutoModelForTokenClassification, AutoModelForImageClassification, AutoModelForImageToImage, AutoModelForImageSegmentation, AutoModelForQuestionAnswering, AutoModelForMaskedLM, AutoTokenizer, AutoModelForMaskGeneration, AutoModelForPreTraining, - AutoModelForTextEncoding, GenerationConfig, - Qwen2VLForConditionalGeneration, AutoImageProcessor, BatchFeature, - T5EncoderModel) + AutoModelForTextEncoding, AutoImageProcessor, BatchFeature, + Qwen2VLForConditionalGeneration, T5EncoderModel) else: print( 'transformer is not installed, please install it if you want to use related modules' @@ -96,18 +96,17 @@ else: if is_transformers_available(): _import_structure['utils.hf_util'] = [ - 'AutoConfig', 'GenerationConfig', 'AutoModel', 'GPTQConfig', - 'Qwen2VLForConditionalGeneration', 'AwqConfig', + 'AutoModel', 'AutoProcessor', 'AutoFeatureExtractor', + 'GenerationConfig', 'AutoConfig', 'GPTQConfig', 'AwqConfig', 'BitsAndBytesConfig', 'AutoModelForCausalLM', - 'AutoModelForSeq2SeqLM', 'AutoTokenizer', - 'AutoModelForSequenceClassification', + 'AutoModelForSeq2SeqLM', 'AutoModelForSequenceClassification', 'AutoModelForTokenClassification', 'AutoModelForImageClassification', 'AutoModelForImageToImage', - 'AutoModelForQuestionAnswering', 'AutoModelForMaskedLM', + 'AutoModelForImageSegmentation', 'AutoModelForQuestionAnswering', + 'AutoModelForMaskedLM', 'AutoTokenizer', 'AutoModelForMaskGeneration', 'AutoModelForPreTraining', - 'AutoModelForTextEncoding', 'AutoModelForTokenClassification', - 'AutoModelForImageSegmentation', 'AutoImageProcessor', - 'BatchFeature', 'T5EncoderModel' + 'AutoModelForTextEncoding', 'AutoImageProcessor', 'BatchFeature', + 'Qwen2VLForConditionalGeneration', 'T5EncoderModel' ] import sys