Audio streaming training with masking

[saeeddhqan]

By utilizing a mask, we can simulate audio streaming.

[farzadab]

Thanks Saeed. I left some comments.

It would be ideal to add some tests, for example to make sure the structure of the mask matches what we expect. If you can think of another test feel free to add one.
I know the repo doesn't have many tests, but we'd like to change that.

[saeeddhqan]

@farzadab Hi,
I created a test file for ultravox_model, moved the function, and changed var names.
I look forward to further improvements.

[farzadab]

Thanks. I left a few nit comments.
Please make sure to run just format to run formatting and use just test to make sure the tests pass.

[saeeddhqan]

@farzadab, I utilized 'just test', and 'just format', and corrected mentioned issues.

[farzadab]

Assuming the tests pass it LGTM. Thanks Saeed!

[farzadab]

Seems like just test failed since mypy is complaining.

I proposed the solution for two of the issues. The last one I couldn't figure out at a glance why the value could be None, but it should be easy to figure out with the code editor's help.

[saeeddhqan]

@farzadab solved.