+ j(&Rt^RIHt^RIHtHt^RIHt^RIH t ^RI H t ^RI H t ^RIHtHt^RIHt^R I Ht] P(!]4t] 'd$]!4'd^R It^R IHt^R IHt^R IHt!RR]4tR #)z3Abstract base class for all Transformers exporters.) annotations)ABCabstractmethod)MutableMapping) TYPE_CHECKING)version)logging)_is_package_availableis_torch_available)ExportConfigMixin)decompose_for_generationN)Cache)GenerationConfig)PreTrainedModelcx]tRt^+t$Rt.tR]R&/tR]R&/tR]R&Rt Rt ] R R l4t RR R llt RtR #) HfExporterz Abstract base class for all Transformers exporters. Subclass and implement [`~HfExporter.export`] to add a new export backend. z list[str]required_packageszdict[str, str] min_versionstested_versionsc &VP4R#)N)validate_environment)selfs&\/home/luca/screen-checking/.venv/lib/python3.14/site-packages/transformers/exporters/base.py__init__HfExporter.__init__8s !!#c  Ra..rCSPFp\VRR7wrgV'gVPV4K-SPP V4pVfKNVR8wgKWVP R^4^,p VP R^4^,p W8wgKVPWYV 34K V'd@RP V3RlV44p \R\S4P R V 24h.p SPP4F}wr]\VRR7wrVR8XgI\P!VP R^4^,4\P!V 48gKcV PV R V R V R 24K V 'd3\\S4P R RP V 4 24hV'dIRP RV44p\P\S4P RV R24R#R#)zYCheck `required_packages` are installed and warn on version drift from `tested_versions`.T)return_versionNzN/A+z, c3x<"TF/qSP9dV RSPV, 2MTxK1 R#5i)z==N)r).0pkgrs& r 2HfExporter.validate_environment..Os>ipbe@T@T9T3%r$..s345Z]]ips7:zTo use z-, please install the following dependencies: z>=z (found )z requires newer versions of: c3:"TFwrq1 RV RV 2xK R#5i)z : installed z , tested N)r r!gotwants& rr"r#^s&e_d^SW[%|C5 $ H_dsu~ is experimental and patches many backend internals; behaviour may differ from what was validated. Version drift detected — z-. If you hit issues, try the tested versions.)rr appendrgetsplitjoin ImportErrortype__name__ritemsrparseloggerwarning)rargskwargsmissingdriftr!exists installedtestedinstalled_base tested_basespecsoutdatedminimum_detailssf*, rrHfExporter.validate_environment;s R))C 5c$ O Fs#))--c2F!i5&8!*a!8!;$ll3215 !0LL#{!CD* IIipET (;(;'<?  rc$V^8dQhRRRRRR/#)modelr sample_inputs)MutableMapping[str, torch.Tensor | Cache]configr r&)formats"r __annotate__HfExporter.__annotate__fs)   A "  rc D\\V4P R24h)u Export the model and return the backend-specific program object. Args: model ([`PreTrainedModel`]): The model to export. sample_inputs (`dict[str, torch.Tensor | Cache]`): **Forward** kwargs — what you'd pass to `model(**sample_inputs)`. These are used directly as the example inputs during tracing. For an autoregressive decode-step export, this means you need to include `past_key_values`, `cache_position`, etc. If you only have generation-style inputs, use [`~HfExporter.export_for_generation`] instead — it runs `model.generate` for you and exports each stage. config ([`~transformers.exporters.configs.ExportConfigMixin`]): Backend-specific configuration. Returns: Backend-specific export artifact. z does not implement `export`. Pick a concrete exporter (`DynamoExporter`, `OnnxExporter`, `ExecutorchExporter`), or override `export` in your subclass with a backend-specific tracing pipeline that consumes `config` and returns the runtime artifact.)NotImplementedErrorr.r/)rrErFrHs&&&&rexportHfExporter.exportes+2"Dz""#$0 0  rNc 0V^8dQhRRRRRRRRR R R R /#) rDrErrFrGrHz0ExportConfigMixin | dict[str, ExportConfigMixin]generation_configzGenerationConfig | Nonemulti_token_decodeboolreturnzdict[str, object]r&)rIs"rrJrKsPHHHAHA H 3 H ! H Hrc  6\VVVVR7p\V\4'dK\V4\V4, pV'd%\ R\ V4 R\ V4 R24hTpM\P Wc4p/p VP4F$wp wrVPWW,R7W&K& V # \dMp \\T4P RT R\T 4P R\T 4 R 24T hR p ? ii;i) u Decompose a generative model and export each component independently. Thin wrapper around [`~exporters.utils.decompose_for_generation`] that calls [`~HfExporter.export`] on every returned `(submodel, forward_inputs)` pair. If you need the intermediate `(submodel, forward_inputs)` pairs (for verification, custom inputs, skipping a stage, …), call [`~exporters.utils.decompose_for_generation`] directly. Args: model ([`PreTrainedModel`]): The generative model to export. Must support `model.generate(**sample_inputs)`. sample_inputs (`dict[str, torch.Tensor | Cache]`): **Generate** kwargs — what you'd pass to `model.generate(**sample_inputs)` (typically `input_ids` + `attention_mask`, plus any modality inputs like `pixel_values` / `input_features` for multi-modal models). Per-stage forward kwargs are captured internally. config ([`~transformers.exporters.configs.ExportConfigMixin`] or `dict[str, ExportConfigMixin]`): Backend-specific configuration. Pass a single config to apply to every component, or a `dict` keyed by component name (e.g. `"image_encoder"`, `"language_model"`, `"lm_head"`, `"decode"`) to override per-component — all component names must be present in the dict. generation_config ([`~generation.GenerationConfig`], *optional*): Forwarded to the `generate()` capture (defaults to the model's own). Pass one with `cache_implementation="static"` to export against a fixed-size `StaticCache`. multi_token_decode (`bool`, *optional*, defaults to `False`): Whether the `decode` component processes multiple query tokens at once — a dynamic query axis (prefill on an empty cache, continuation-from-past otherwise) — vs the classic single-token step (see [`~exporters.utils.decompose_for_generation`]). Only stays dynamic under a dynamic-shape export (`config.dynamic=True`). Returns: `dict[str, Any]`: `{component_name: backend_specific_artifact}` — same keys as [`~exporters.utils.decompose_for_generation`]. Values are whatever [`~HfExporter.export`] returns for the concrete backend (`ExportedProgram`, `ONNXProgram`, `ExecutorchProgramManager`). )rQrRz4Per-component `config` dict is missing entries for: z$. Expected one entry per component: .)rHz.export failed on component 'z ' (submodel=z , input keys=z).N)r isinstancedictset ValueErrorsortedfromkeysr0rN Exception RuntimeErrorr.r/list)rrErFrHrQrR componentsr6configsexportednamesubmodel subinputses&&&&&& rexport_for_generation HfExporter.export_for_generations.X.  /1   fd # #*oF 3G J6RY?J[\99? 9K8LAOGmmJ7G&(+5+;+;+= 'D'8 !%X!W,>  "Dz**++HO!!%h!8!8 9tIFWWY[ s C D ADDr&)NF)r/ __module__ __qualname____firstlineno____doc__r__annotations__rrrrrrNrg__static_attributes__r&rrrr+sS $&y%#%L.%&(O^($(T  >HHrr)rl __future__rabcrrcollections.abcrtypingr packagingrutilsrutils.import_utilsr r rar r get_loggerr/r2torch cache_utilsr generationrmodeling_utilsrrr&rrr{s_:"#* J&+   H %'14bbr