search

aiola-lab / whisper-medusa

Whisper with Medusa heads
MIT License
773 stars 47 forks source link

Plans to release training codebase? 🤗 #17

Open Vaibhavs10 opened 1 month ago

Vaibhavs10 commented 1 month ago

Hi again, VB from Hugging Face here, I was wondering if y'all have plans to release the training codebase as well!

Given Medusa heads don't perform as well on out-of-distribution data, and the community can potentially train more diverse heads?

Would be keen to collab on that! let me know what you think!

Cheers, VB

AvivSham commented 1 month ago

Hi @Vaibhavs10, Thank you for your interest in our work! Indeed we are planning to release training codebase in the near future. Regarding collaboration, I will send you a private message via email soon.

Vaibhavs10 commented 1 month ago

Hey @AvivSham - perfect! I'm at Vaibhav [at] huggingface [dot] co 🤗

AvivSham commented 1 month ago

@Vaibhavs10 you should see it in your inbox 🤗