Skip to content

Latest commit

History

History
39 lines (22 loc) 路 1.46 KB

dummy_input_generators.mdx

File metadata and controls

39 lines (22 loc) 路 1.46 KB

Dummy Input Generators

It is very common to have to generate dummy inputs to perform a task (tracing, exporting a model to some backend, testing model outputs, etc). The goal of [~optimum.utils.input_generators.DummyInputGenerator] classes is to make this generation easy and re-usable.

Base class

[[autodoc]] optimum.utils.input_generators.DummyInputGenerator

Existing dummy input generators

[[autodoc]] optimum.utils.input_generators.DummyTextInputGenerator

[[autodoc]] optimum.utils.input_generators.DummyDecoderTextInputGenerator

[[autodoc]] optimum.utils.input_generators.DummyPastKeyValuesGenerator

[[autodoc]] optimum.utils.input_generators.DummySeq2SeqPastKeyValuesGenerator

[[autodoc]] optimum.utils.input_generators.DummyBboxInputGenerator

[[autodoc]] optimum.utils.input_generators.DummyVisionInputGenerator

[[autodoc]] optimum.utils.input_generators.DummyAudioInputGenerator