
    ujs                        d dl Z d dlmZ d dlmZmZ er.ddlmZmZm	Z	 ddl
mZ ddlmZmZ ddlmZmZ dd	lmZmZ dd
lmZmZ ddlmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2 ddl3m4Z4m5Z5 ddl6m7Z7 ddl8m9Z9m:Z: ddl;m<Z< ddl=m>Z>m?Z?m@Z@mAZAmBZB ddlCmDZD ddlEmFZFmGZGmHZH  e            r_ddlEmIZImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSmTZTmUZUmVZVmWZWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^m_Z_m`Z`maZambZbmcZcmdZdmeZemfZfmgZgmhZhmiZimjZjmkZkmlZlmmZmmnZnmoZompZpmqZqmrZrmsZsmtZtmuZu n evd           ddlwmxZxmyZy ddlzm{Z{ ddl|m}Z}m~Z~ dS i dddgdg ddg dddgd d!d"gd#d$d%gd&d'd(gd)d*d+gd,g d-d.d/d0gd1d2gd3d4d5gd6d7d8gd9d:gd;d<gd=d>gZd d?lmZ d d@lmZ i Z ee          ZdA eD             ZeD ]Z eee          ee<   dB Zd dlZ ee e            dC         eeeeD          ej        e<   dS )E    N)TYPE_CHECKING)LazyImportModuleis_transformers_available   )ExporterTfModelExporterTorchModelExporter)HubApi)check_local_model_is_latestcheck_model_is_id)push_to_hubpush_to_hub_async)snapshot_downloaddataset_snapshot_download)model_file_downloaddataset_file_download)AccuracyMetricAudioNoiseMetric
BleuMetricImageColorEnhanceMetricImageColorizationMetricImageDenoiseMetricImageInpaintingMetric#ImageInstanceSegmentationCOCOMetricImagePortraitEnhancementMetric'ImageQualityAssessmentDegradationMetricImageQualityAssessmentMosMetric
LossMetricMetricMovieSceneSegmentationMetricOCRRecognitionMetric	PplMetric&ReferringVideoObjectSegmentationMetricSequenceClassificationMetricTextGenerationMetricTextRankingMetricTokenClassificationMetricVideoFrameInterpolationMetricVideoStabilizationMetricVideoSummarizationMetricVideoSuperResolutionMetrictask_default_metrics)Model
TorchModel)	MsDataset)Pipelinepipeline)Preprocessor)EpochBasedTrainerHookPriorityTrainingArgsbuild_dataset_from_file)Tasks)	patch_hubpatch_contextunpatch_hub)-	AutoModelAutoProcessorAutoFeatureExtractorGenerationConfig
AutoConfig
GPTQConfig	AwqConfigBitsAndBytesConfigAutoModelForCausalLMAutoModelForSeq2SeqLMAutoModelForVision2Seq"AutoModelForSequenceClassificationAutoModelForTokenClassificationAutoModelForImageClassificationAutoModelForImageTextToText'AutoModelForZeroShotImageClassificationAutoModelForKeypointDetection%AutoModelForDocumentQuestionAnswering AutoModelForSemanticSegmentation!AutoModelForUniversalSegmentation AutoModelForInstanceSegmentationAutoModelForObjectDetection#AutoModelForZeroShotObjectDetectionAutoModelForAudioClassificationAutoModelForSpeechSeq2SeqAutoModelForMaskedImageModeling#AutoModelForVisualQuestionAnswering"AutoModelForTableQuestionAnsweringAutoModelForImageToImageAutoModelForImageSegmentationAutoModelForQuestionAnsweringAutoModelForMaskedLMAutoTokenizerAutoModelForMaskGenerationAutoModelForPreTrainingAutoModelForTextEncodingAutoImageProcessorBatchFeatureQwen2VLForConditionalGenerationT5EncoderModel"Qwen2_5_VLForConditionalGeneration
LlamaModelLlamaPreTrainedModelLlamaForCausalLMhf_pipelinezRtransformer is not installed, please install it if you want to use related modules)create_model_if_not_existread_config)
get_logger)__release_datetime____version__versionrl   rm   trainers)r3   r6   r4   r5   r7   	exporterszhub.apir
   zhub.snapshot_downloadr   r   zhub.file_downloadr   r   zhub.push_to_hubr   r   zhub.check_modelr   r   metrics)r   r   r,   r   r   r   r   r$   r%   r'   r*   r    r   r   r   r#   r(   r)   r+   r"   r   r   r&   r   r   r!   modelsr-   r.   preprocessorsr2   	pipelinesr0   r1   z	utils.hubrj   ri   zutils.loggerrk   zutils.constantr8   
msdatasetsr/   )hf_util)_patch_pretrained_classc                 <    g | ]}|                     d           |S )__)
startswith).0attrs     c/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/modelscope/__init__.py
<listcomp>r~   u   s)    HHH$//$2G2GHtHHH    c                 <   g d}d }d }|D ]8}	 t          t          j        |          |           }|} n# t          $ r Y 5w xY w|t	          d|  d| d          |dk    r"t          |          rddlm}  ||           S |S t          |gd	          }|d         S )
N)transformerspeft	diffuserskernelsz"Cannot import available module of z$ in modelscope, or related packages()r   r   )_wrap_kernels_callableT)wrap)	getattr	importlibimport_module	ExceptionAttributeErrorcallable modelscope.utils.hf_util.patcherr   rw   )namehf_pkgsmodulematched_pkgpkgr   s         r}   try_import_from_hfr   y   s   BBB 	 	C !8!=!=tDD!    > 3T 3 3(/3 3 34 4 4 )## 4+ + + + + +--d333M(&===ays   $4
A A__file__)module_specextra_objectsextra_import_func)r   typingr   modelscope.utils.import_utilsr   r   rp   r   r   r	   hub.apir
   hub.check_modelr   r   hub.push_to_hubr   r   hub.snapshot_downloadr   r   hub.file_downloadr   r   rq   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   rr   r-   r.   ru   r/   rt   r0   r1   rs   r2   ro   r3   r4   r5   r6   r7   utils.constantr8   utils.hf_utilr9   r:   r;   r<   r=   r>   r?   r@   rA   rB   rC   rD   rE   rF   rG   rH   rI   rJ   rK   rL   rM   rN   rO   rP   rQ   rR   rS   rT   rU   rV   rW   rX   rY   rZ   r[   r\   r]   r^   r_   r`   ra   rb   rc   rd   re   rf   rg   rh   print	utils.hubri   rj   utils.loggerrk   rn   rl   rm   _import_structuremodelscope.utilsrv   r   rw   r   dir
attributesimports_importr   r   sys__name__globals__spec__modules r   r}   <module>r      s                  F F F F F F F F  UHHHHHHHHHHOOOOOOOO????????SSSSSSSSMMMMMMMM                                                        *)))))))%%%%%%--------++++++4 4 4 4 4 4 4 4 4 4 4 4 4 4%%%%%%DDDDDDDDDD  "" 
	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A 	A2 	`	
 	
 	
 BAAAAAAA((((((::::::::::(*M:( 
 
 
( 	 
 
 
( 	H:( 	 	9:( 	35LM( 	M+>?(  		;<#($ 	 
 
 
%(B 	7L)C(D 	.)E(F 	j*-G(H 	m%@AI(J 	K(L 	7)M(N 	{mO(T )(((((HHHHHHMWJHH
HHHG ; ;!('!:!:g  6 JJJ,,		*#,  CKr   