9i+SSKrSSKJr SSKJrJr SSKJrJ r J r J r J r SSK Jr SSKJr SSKJrJr SSKJrJrJr SSKJrJr SS KJrJrJr SS KJ r SS K!J"r" SS K#J$r$J%r% \""5r&\ S r'"SS\5r(g)N)ABCabstractmethod)AnyDictListOptionalUnion)snapshot_download)Tasks)build_backbone build_model)can_load_by_mscheck_model_from_owner_grouptry_to_load_hf_model)Config ConfigDict)DEFAULT_MODEL_REVISIONInvoke ModelFile) verify_device) get_logger)register_modelhub_reporegister_plugins_repo)z torch.Tensorz tf.Tensorc F\rSrSrSrSrS\\\44Sjr SS\ \S\ \4S jjr \ S\\\44S j5r S \\\4S\\\44S jr\S 5r\\SSS4S\S\ \S\S\S\ \4 Sjj5rSS\\\R.4S\\\\4S\ \4SjjrSrg)ModelzBase model interface. cXlURSS5n[U5 X@lURSS5Ulg)Ndevicegputrust_remote_codeF) model_dirgetr _device_namer )selfr!argskwargs device_names a/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/modelscope/models/base/base_model.py__init__Model.__init__s9"jj51 k"'!',?!GreturncDURUR"U0UD65$)N) postprocessforwardr$r%r&s r(__call__Model.__call__$s! d =f =>>r+Ninfo_strr!chU=(d Sn[US9(dUR(dU5egg)aCheck trust_remote_code if the model needs to import extra libs Args: info_str(str): The info showed to user if trust_remote_code is `False`. model_dir(`Optional[str]`): The local model directory. If is a trusted model, check remote code will pass. zThis model requires `trust_remote_code` to be `True` because it needs to import extra libs or execute the code in the model repo, setting this to true means you trust the files in it.)r!N)rr )r$r3r!s r(check_trust_remote_codeModel.check_trust_remote_code's:0 / ,i@)) 38 3)Ar+cg)zd Run the forward pass for a model. Returns: Dict[str, Any]: output from the model forward pass Nr0s r(r/ Model.forward7s r+inputsc U$)zModel specific postprocess and convert model output to standard model outputs. Args: inputs: input data Return: dict of results: a dict containing outputs of model, each output should have the standard output name. r8)r$r:r&s r(r.Model.postprocessAs  r+c U"S0UD6$)zDefine the instantiation method of a model,default method is by calling the constructor. Note that in the case of no loading model process in constructor of a task model, a load_model method is added, and thus this method is overloaded r8r8)clsr&s r( _instantiateModel._instantiateNs}V}r+Fmodel_name_or_pathrevisioncfg_dictrr c rURS5nUbURS5 UR[R5nUb UR[R5 O[RnURSS5n [ R "U5(aUn O3USLa [S5e[R<SU<3n[UUUU S9n [RSU 35 [ R"U [R5n Sn UbUn O:[RR U 5(a[ R""U 5n [%U S S5n S U;aURS 5n [%U S ['55n[)US 5(a"[)US 5(dUR*Ul[%US S5n[/U[05(aUR3S 5(aSUSS-nURSS5nUc[5XU5(aSnSnUS;a [7XU40UD6nUb+URSS5nUcUbUR9U5nU$U c[;S[RS35eXlU R?S5nU(aU(d [S5eU(aU(a[RAS5 [CXS9 0nU(aSU0n[EU5 URG5H unnUUU'M UbXNl$U [JRLLaSUl'[QU5nO [SXUS9n[)U S5(aU RTUl*[)US5(dU Ul+URSS5 UUl,U UlU$)aInstantiate a model from local directory or remote model repo. Note that when loading from remote, the model revision can be specified. Args: model_name_or_path(str): A model dir or a model id to be loaded revision(str, `optional`): The revision used when the model_name_or_path is a model id of the remote hub. default `master`. cfg_dict(Config, `optional`): An optional model config. If provided, it will replace the config read out of the `model_name_or_path` device(str, `optional`): The device to load the model. trust_remote_code(bool, `optional`): Whether to trust and allow execution of remote code. Default is False. **kwargs: task(str, `optional`): The `Tasks` enumeration value to replace the task value read out of config in the `model_name_or_path`. This is useful when the model to be loaded is not equal to the model saved. For example, load a `backbone` into a `text-classification` model. Other kwargs will be directly fed into the `model` key, to replace the default configs. use_hf(bool, `optional`): If set to True, it will initialize the model using AutoModel or AutoModelFor* from hf. If set to False, the model is loaded using the modelscope mode. If set to None, the loading mode will be automatically selected. ignore_file_pattern(List[str], `optional`): This parameter is passed to snapshot_download device_map(str | Dict[str, str], `optional`): This parameter is passed to AutoModel or AutoModelFor* torch_dtype(torch.dtype, `optional`): This parameter is passed to AutoModel or AutoModelFor* config(PretrainedConfig, `optional`): This parameter is passed to AutoModel or AutoModelFor* Returns: A model instance. Examples: >>> from modelscope.models import Model >>> Model.from_pretrained('damo/nlp_structbert_backbone_base_std', task='text-classification') model_prefetchedNignore_file_patternTz9Expecting model is pre-fetched locally, but is not found./) user_agentrFzinitialize model from taskmodel model_typetypercudause_hfF>NT device_map`z` file not found.pluginsaDetected plugins field in the model configuration file, but trust_remote_code=True was not explicitly set. To prevent potential execution of malicious code, loading has been refused. If you trust this model repository, please pass trust_remote_code=True to from_pretrained.zUse trust_remote_code=True. Will invoke codes or install plugins from remote model repo. Please make sure that you can trust the external codes.) allow_remoter ) task_name default_argspipelinecfgr!)-r"poprKEY PRETRAINEDospexists RuntimeErrorr loggerinfojoinr CONFIGURATIONospathr from_filegetattrrhasattrrKrL isinstancestr startswithrrtoFileNotFoundErrorr!safe_getwarningrritemsrr backbone init_backboner r rVrWname)r>rArBrCrr r& prefetched invoked_byrFlocal_model_dirconfiguration_pathrWrT model_cfgrKrOrJrPrRrUkvs r(from_pretrainedModel.from_pretrainedWsvXZZ 23  ! JJ) *ZZ +  ! JJvzz "**J$jj)>E ::( ) )0OT!"O%+JJ ;J/"%$7 9O  ,_,=>? XXoy7N7NO  C WW^^. / /""#56CC. V  6*IC*,7 9l + +GIv4N4N&11INY5 fc " "v'8'8'?'?fQRj(FHd+ >n_-799F \ !(V3+13E  L$7J!f&8(L ;#I++,,=>@ @-,,y) ,m  ( NNJ K O /1BCLg&LLNDAqIaL#  %   &&*I #"9-E\KE 3 # # \\ENue$$EI k4(' ) r+ target_foldersave_checkpoint_namesconfigc [S5e)asave the pretrained model, its configuration and other related files to a directory, so that it can be re-loaded Args: target_folder (Union[str, os.PathLike]): Directory to which to save. Will be created if it doesn't exist. save_checkpoint_names (Union[str, List[str]]): The checkpoint names to be saved in the target_folder config (Optional[dict], optional): The config for the configuration.json, might not be identical with model.config z>save_pretrained method need to be implemented by the subclass.)NotImplementedError)r$r{r|r}r&s r(save_pretrainedModel.save_pretraineds$" LN Nr+)r#r!r )NN)__name__ __module__ __qualname____firstlineno____doc__r)rrhrr1rr5rr/r. classmethodr?rrboolryr rbPathLikerdictr__static_attributes__r8r+r(rrsSH?4S>?;?;?4*23-4+3C=4  $sCx.   $sCx. tCH~ 3I+/&* NNdHL15N',S"++-='>N/4S$s)^/DN!)NNr+r))rbos.pathrcr[abcrrtypingrrrrr modelscope.hub.snapshot_downloadr modelscope.metainfor modelscope.models.builderr r modelscope.utils.automodel_utilsrrrmodelscope.utils.configrrmodelscope.utils.constantrrrmodelscope.utils.devicermodelscope.utils.loggerrmodelscope.utils.pluginsrrr^Tensorrr8r+r(rsc #33>%ADD7OO1.=  * +bNCbNr+