PABannier / bark.cpp

Suno AI's Bark model in C/C++ for fast text-to-speech
MIT License
693 stars 56 forks source link

ENH/MNT Deterministic tests for all encoders #34

Closed PABannier closed 1 year ago

PABannier commented 1 year ago

Currently, the unit tests for all the encoders are stochastic as they depend on gpt_sample which is random via the Mersenne twister. To make tests reproducible, we sample by taking the argmax of the logits.

Note that we can't fix the seed for both Python bark and bark.cpp.