Where can I find all model-specific parameters that can be passed to this model in the from_pretrained method?

#154

by denis-kazakov - opened Sep 8

Sep 8

The model card example includes low_cpu_mem_usage and use_safetensors, but what are others?
help(AutoModelForSpeechSeq2Seq.from_pretrained) only shows general parameters.

nbroad

Sep 8

•

edited Sep 8

https://github.com/huggingface/transformers/blob/66bc4def9505fa7c7fe4aa7a248c34a026bb552b/src/transformers/modeling_utils.py#L2943C9-L2943C24

and

https://github.com/huggingface/transformers/blob/66bc4def9505fa7c7fe4aa7a248c34a026bb552b/src/transformers/models/whisper/configuration_whisper.py#L59

denis-kazakov

Sep 9

Thank you!

denis-kazakov changed discussion status to closed Sep 9

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment