Skip to main content

Transformers Engine Schema

Engine version: 4.57.3
Discovered at: 2026-05-06T22:57:22+02:00
Discovery method: inspect.signature(from_pretrained) + GenerationConfig().to_dict()
Schema version: 1.0.0

Summary: 9 engine parameters, 67 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 has no type annotations; None defaults yield type='unknown' Affected fields: max_new_tokens, min_new_tokens, max_time, stop_strings, cache_implementation, cache_config, return_legacy_cache, prefill_chunk_size, min_p, bad_words_ids, forced_bos_token_id, forced_eos_token_id, exponential_decay_length_penalty, suppress_tokens, begin_suppress_tokens, sequence_bias, guidance_scale, watermarking_config, output_logits, pad_token_id, bos_token_id, eos_token_id, decoder_start_token_id, prompt_lookup_num_tokens, max_matching_ngram_size, assistant_early_exit, low_memory, penalty_alpha, dola_layers, constraints, force_words_ids

Engine Parameters

FieldTypeDefaultDescription
config`PretrainedConfigstrPathLike
cache_dir`strPathLikeNone`
ignore_mismatched_sizesboolfalse
force_downloadboolfalse
local_files_onlyboolfalse
token`strboolNone`
revisionstrmain
use_safetensors`boolNone`
weights_onlybooltrue

Sampling Parameters

FieldTypeDefaultDescription
max_lengthint20
max_new_tokensunknown
min_lengthint0
min_new_tokensunknown
early_stoppingboolfalse
max_timeunknown
stop_stringsunknown
do_sampleboolfalse
num_beamsint1
use_cachebooltrue
cache_implementationunknown
cache_configunknown
return_legacy_cacheunknown
prefill_chunk_sizeunknown
temperaturefloat1.0
top_kint50
top_pfloat1.0
min_punknown
typical_pfloat1.0
epsilon_cutofffloat0.0
eta_cutofffloat0.0
repetition_penaltyfloat1.0
encoder_repetition_penaltyfloat1.0
length_penaltyfloat1.0
no_repeat_ngram_sizeint0
bad_words_idsunknown
renormalize_logitsboolfalse
forced_bos_token_idunknown
forced_eos_token_idunknown
remove_invalid_valuesboolfalse
exponential_decay_length_penaltyunknown
suppress_tokensunknown
begin_suppress_tokensunknown
sequence_biasunknown
token_healingboolfalse
guidance_scaleunknown
watermarking_configunknown
num_return_sequencesint1
output_attentionsboolfalse
output_hidden_statesboolfalse
output_scoresboolfalse
output_logitsunknown
return_dict_in_generateboolfalse
pad_token_idunknown
bos_token_idunknown
eos_token_idunknown
encoder_no_repeat_ngram_sizeint0
decoder_start_token_idunknown
is_assistantboolfalse
num_assistant_tokensint20
num_assistant_tokens_schedulestrconstant
assistant_confidence_thresholdfloat0.4
prompt_lookup_num_tokensunknown
max_matching_ngram_sizeunknown
assistant_early_exitunknown
assistant_lookbehindint10
target_lookbehindint10
disable_compileboolfalse
low_memoryunknown
penalty_alphaunknown
dola_layersunknown
diversity_penaltyfloat0.0
num_beam_groupsint1
constraintsunknown
force_words_idsunknown
_from_model_configboolfalse
transformers_versionstr4.57.3