Minor patch, couple imports missing

lucidrains / audiolm-pytorch

Implementation of AudioLM, a SOTA Language Modeling Approach to Audio Generation out of Google Research, in Pytorch

MIT License

2.41k stars 256 forks source link

Closed tchamb closed 1 year ago

tchamb commented 1 year ago

Minor bugs

functools, einops, and torch import missing from encodec.py
Coarse transformer trainer was saving its checkpoints with "fine" in the filename

lucidrains commented 1 year ago

@tchamb thank you!