o
    rqi>                     @   s  d dl Z d dlmZ d dlmZmZ er6d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 ZeeZdAdB eD ZeD ]
Zeeeee< qdCdD Zd dlZeee dE eeeedFeje< dS )G    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__versionrk   rl   trainers)r2   r5   r3   r4   r6   	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-   preprocessorsr1   	pipelinesr/   r0   z	utils.hubri   rh   zutils.loggerrj   zutils.constantr7   
msdatasetsr.   )hf_util)_patch_pretrained_classc                 C   s   g | ]	}| d s|qS )__)
startswith).0attr r{   Z/home/app/PaddleOCR-VL/.venv_paddleocr/lib/python3.10/site-packages/modelscope/__init__.py
<listcomp>u   s    r}   c              	   C   sr   g d}d }|D ]}zt t|| }W  n	 ty   Y qw |d ur.t|gdd}|d S td|  d| d)N)ZtransformersZpeftZ	diffusersT)wrapz"Cannot import available module of z$ in modelscope, or related packages()r   )getattr	importlibimport_module	Exceptionrv   ImportError)nameZhf_pkgsmodulepkgr{   r{   r|   try_import_from_hfy   s"   r   __file__)Zmodule_specextra_objectsZextra_import_func)r   typingr   Zmodelscope.utils.import_utilsr   r   ro   r   r   r   Zhub.apir	   Zhub.check_modelr
   r   Zhub.push_to_hubr   r   Zhub.snapshot_downloadr   r   Zhub.file_downloadr   r   rp   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+   rq   r,   r-   rt   r.   rs   r/   r0   rr   r1   rn   r2   r3   r4   r5   r6   Zutils.constantr7   Zutils.hf_utilr8   r9   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   printZ	utils.hubrh   ri   Zutils.loggerrj   rm   rk   rl   Z_import_structureZmodelscope.utilsru   Z modelscope.utils.hf_util.patcherrv   r   dir
attributesZimportsZ_importr   r   sys__name__globals__spec__modulesr{   r{   r{   r|   <module>   s   p!"#$%&'*