Downloads · 30 days
746
0% of all-time downloads
nikita-savelyev-intel/tiny-random-whisper
tiny-random-whisper is a machine learning model from nikita-savelyev-intel. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
Downloads · 30 days
746
0% of all-time downloads
All-time downloads
174K
Public
Parameters
3M
11.8 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors11.8 MB · 86%
From the Hugging Face model README
Obtained with
from transformers import WhisperForConditionalGeneration, WhisperConfig, WhisperProcessor, GenerationConfig
model_id = "openai/whisper-tiny"
orig_config = WhisperConfig.from_pretrained(model_id)
small_config = dict(
d_model=48,
encoder_layers=2,
decoder_layers=2,
encoder_attention_heads=3,
decoder_attention_heads=3,
encoder_ffn_dim=768,
decoder_ffn_dim=768,
)
small_config = WhisperConfig(**{**orig_config.to_dict(), **small_config})
save_dir = "tiny-random-whisper"
WhisperForConditionalGeneration(small_config).save_pretrained(save_dir)
WhisperProcessor.from_pretrained(model_id).save_pretrained(save_dir)
GenerationConfig.from_pretrained(model_id).save_pretrained(save_dir)