Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 12:10:09 +01:00
Code Issues Packages Projects Releases Wiki Activity

rearrange

Browse Source
This commit is contained in:
Yingda Chen
2024-11-28 19:32:10 +08:00
parent 7af4ef0f90
commit 089b5a327f
1 changed files with 12 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
modelscope/__init__.py
View File

@@ -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