Gadersd / whisper-burn

A Rust implementation of OpenAI's Whisper model using the burn framework

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Tinygrad dependancy for weights conversion

DavidGOrtega opened this issue · comments

May I ask why the need of Tinygrad for the weights conversions? The script seems to be dumping them with np afterwards is read by tinygrad.

I used the tinygrad whisper implementation as a reference when porting since it's concise. There's no reason the script couldn't be easily modified to only use torch for loading the weights.