search

PABannier / bark.cpp

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

FIX Encodec tests #87

Closed PABannier closed 10 months ago

PABannier commented 10 months ago

Currently, out of the 3 Encodec tests, 2 pass and one fail:

bark_forward_encodec: mem per token =     0.00 MB
bark_forward_encodec:  predict time =  1130.08 ms
bark_forward_encodec:    total time =  1198.17 ms

main: ./data/encodec/test_pass_encodec_1.bin
main:     test 1 passed.

bark_forward_encodec: mem per token =     0.00 MB
bark_forward_encodec:  predict time =  3756.10 ms
bark_forward_encodec:    total time =  3796.49 ms

main: ./data/encodec/test_pass_encodec_2.bin
main:     test 2 passed.

bark_forward_encodec: mem per token =     0.00 MB
bark_forward_encodec:  predict time =   681.15 ms
bark_forward_encodec:    total time =   770.30 ms

main: ./data/encodec/test_pass_encodec_3.bin
run_test: abs_tol=0.0010, abs max viol=38923531967525382011995815936.0000, viol=100.0%
main:     test 3 failed.

Oddly enough, when I only run test 3, it passes without any problem. 

bark_forward_encodec: mem per token =     0.00 MB
bark_forward_encodec:  predict time =   572.53 ms
bark_forward_encodec:    total time =   632.48 ms

main: ./data/encodec/test_pass_encodec_3.bin
main:     test 1 passed.

I suspect there's a problem when cleaning the memory ctxs.