6pj-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%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)filter_plugin_in_whitelistregister_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@lURSS5=(d [ U5Ulg)Ndevicegputrust_remote_codeF) model_dirgetr _device_namerr!)selfr"argskwargs device_names ]/var/www/html/pdf-tiff/venv/lib/python3.13/site-packages/modelscope/models/base/base_model.py__init__Model.__init__sL"jj51 k"'!'  "">29= returncDURUR"U0UD65$)N) postprocessforwardr%r&r's r)__call__Model.__call__'s! d =f =>>r,Ninfo_strr"cpU=(d Sn[US9(dUR(d [U5egg)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! RuntimeError)r%r4r"s r)check_trust_remote_codeModel.check_trust_remote_code*s?0 / ,i@))"8,,*Ar,cg)zd Run the forward pass for a model. Returns: Dict[str, Any]: output from the model forward pass Nr1s r)r0 Model.forward<s 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. r:)r%r<r's r)r/Model.postprocessFs  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 r:r:)clsr's r) _instantiateModel._instantiateSs}V}r,Fmodel_name_or_pathrevisioncfg_dictrr!c URS5nUbURS5 UR[R5nUb UR[R5 O[Rn[ U5n U=(d U nURSS5n [ R"U5(aUn O3USLa [S5e[R<SU<3n[UUUU S9n [RSU 35 [ R"U [R5n Sn UbUn O:[R RU 5(a["R$"U 5n ['U S S5nS U;aURS 5n['U S [)55n[+US 5(a"[+US 5(dUR,Ul['US S5n[1U[25(aUR5S 5(aSUSS-nURSS5nUc[7XU5(aSnSnUS;a [9XU40UD6nUb+URSS5nUcUbUR;U5nU$U c[=S[RS35eXlU RAS5n[CU5(aU(d [S5eU(aU(a[RES5 [GXS9 0nU(a U (dSU0n[IU5 URK5H unnUUU'M UbXOl&U[NRPLaSUl)[UU5nO [WXUS9n[+U S5(aU RXUl,[+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_agentrHzinitialize model from taskmodel model_typetyper cudause_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 PRETRAINEDrospexistsr6r loggerinfojoinr CONFIGURATIONospathr from_filegetattrrhasattrrMrN isinstancestr startswithrrtoFileNotFoundErrorr"safe_getrwarningrritemsrr backbone init_backboner r rXrYname)r@rCrDrErr!r' prefetched invoked_by_model_trustedrHlocal_model_dirconfiguration_pathrYrV model_cfgrMrQrLrRrTrWkvs r)from_pretrainedModel.from_pretrained\sXZZ 23  ! JJ) *ZZ +  ! JJvzz "**J56HI-?$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) %g . .7Hm  ( 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~rr's r)save_pretrainedModel.save_pretraineds$" LN Nr,)r$r"r!)NN)__name__ __module__ __qualname____firstlineno____doc__r*rrirr2rr7rr0r/ classmethodrArrboolr{r rcPathLikerdictr__static_attributes__r:r,r)rrsR>?4S>?;?;?-*23--+3C=-$ $sCx.   $sCx. tCH~ 3I+/&* OOfHL15N',S"++-='>N/4S$s)^/DN!)NNr,r)*rcos.pathrdr]abcrrtypingrrrrr modelscope.hub.snapshot_downloadr modelscope.metainfor modelscope.models.builderr r modelscope.utils.automodel_utilsrrrmodelscope.utils.configrrmodelscope.utils.constantrrrmodelscope.utils.devicermodelscope.utils.loggerrmodelscope.utils.pluginsrrrr_Tensorrr:r,r)rsh #33>%ADD7OO1.==  * +gNCgNr,