Skip to main content

Transformers Engine Schema

Engine version: 5.14.1
Discovered at: 2026-08-06T13:06:39.990799+00:00
Discovery method: inspect.signature(from_pretrained) + GenerationConfig().to_dict() with docstring-Args type recovery
Schema version: 1.0.0

Summary: 11 engine parameters, 72 sampling parameters.

Discovery limitations

  • engine_params - from_pretrained accepts **kwargs; kwargs are not in the signature (documented kwargs live in the class docstring only) Affected fields: AutoModelForCausalLM.from_pretrained.**model_args, AutoModelForCausalLM.from_pretrained.**kwargs, PreTrainedModel.from_pretrained.**model_args, PreTrainedModel.from_pretrained.**kwargs
  • sampling_params - GenerationConfig field is None-defaulted and undocumented in the docstring Args block; type='unknown' Affected fields: use_mtp, cache_config, bad_words_ids, exponential_decay_length_penalty, suppress_tokens, begin_suppress_tokens, sequence_bias, watermarking_config, eos_token_id, compile_config, continuous_batching_config, low_memory, penalty_alpha, dola_layers, diversity_penalty, num_beam_groups, constraints, force_words_ids, prefill_chunk_size, _from_model_config

Engine Parameters

FieldTypeDefaultDescription
config`PreTrainedConfigstrPathLike
cache_dir`strPathLikeNone`
ignore_mismatched_sizesboolfalse
force_downloadboolfalse
local_files_onlyboolfalse
token`strboolNone`
revisionstrmain
use_safetensors`boolNone`-
weights_onlybooltrue
fusion_config`dict[str, booldict[str, Any]]None`
disable_mmap`boolNone`-

Sampling Parameters

FieldTypeDefaultDescription
max_lengthint-
max_new_tokensint-
min_lengthint-
min_new_tokensint-
early_stopping`boolLiteral['never']`-
max_timefloat-
stop_stringsstr-
do_samplebool-
num_beamsint-
use_mtpunknown-
use_cachebool-
cache_implementationstr-
cache_configunknown-
max_cache_lenint-
temperaturefloat-
top_kint-
top_pfloat-
min_pfloat-
top_hfloat-
typical_pfloat-
epsilon_cutofffloat-
eta_cutofffloat-
repetition_penaltyfloat-
encoder_repetition_penaltyfloat-
length_penaltyfloat-
no_repeat_ngram_sizeint-
bad_words_idsunknown-
renormalize_logitsbool-
forced_bos_token_idint-
forced_eos_token_idint-
remove_invalid_valuesbool-
exponential_decay_length_penaltyunknown-
suppress_tokensunknown-
begin_suppress_tokensunknown-
sequence_biasunknown-
token_healingbool-
guidance_scalefloat-
watermarking_configunknown-
num_return_sequencesint-
output_attentionsbool-
output_hidden_statesbool-
output_scoresbool-
output_logitsbool-
return_dict_in_generatebool-
pad_token_idint-
bos_token_idint-
eos_token_idunknown-
encoder_no_repeat_ngram_sizeint-
decoder_start_token_idint-
is_assistantbool-
num_assistant_tokensint-
num_assistant_tokens_schedulestr-
assistant_confidence_thresholdfloat-
prompt_lookup_num_tokensint-
max_matching_ngram_sizeint-
assistant_early_exitint-
assistant_lookbehindint-
target_lookbehindint-
assistant_ensemble_weightfloat-
compile_configunknown-
disable_compilebool-
continuous_batching_configunknown-
low_memoryunknown-
penalty_alphaunknown-
dola_layersunknown-
diversity_penaltyunknown-
num_beam_groupsunknown-
constraintsunknown-
force_words_idsunknown-
prefill_chunk_sizeunknown-
_from_model_configunknown-
transformers_versionstr5.14.1