diff --git a/audiocraft/__pycache__/__init__.cpython-310.pyc b/audiocraft/__pycache__/__init__.cpython-310.pyc
index 0e2979ad3de3d4b198c99910cf462aa4fa95645a..895ed9ab3a31e060453dfaca40434f517c9ea9ae 100644
Binary files a/audiocraft/__pycache__/__init__.cpython-310.pyc and b/audiocraft/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/__pycache__/environment.cpython-310.pyc b/audiocraft/__pycache__/environment.cpython-310.pyc
index 821a1853b1078a72b251edbed0a59eaa4fc3f734..f41ed26a7a3f593f327d329f15b348cef3142512 100644
Binary files a/audiocraft/__pycache__/environment.cpython-310.pyc and b/audiocraft/__pycache__/environment.cpython-310.pyc differ
diff --git a/audiocraft/adversarial/__pycache__/__init__.cpython-310.pyc b/audiocraft/adversarial/__pycache__/__init__.cpython-310.pyc
index 076bfe1b479296918dc8056bfe8369d69e4334de..97989ca816af970a2e8b937358c56b87a8577dd0 100644
Binary files a/audiocraft/adversarial/__pycache__/__init__.cpython-310.pyc and b/audiocraft/adversarial/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/adversarial/__pycache__/losses.cpython-310.pyc b/audiocraft/adversarial/__pycache__/losses.cpython-310.pyc
index 4c20cce860fb9d6b5a64b8d11f268112cc64487e..301211b70e061429a84fef7265b97a63992965b4 100644
Binary files a/audiocraft/adversarial/__pycache__/losses.cpython-310.pyc and b/audiocraft/adversarial/__pycache__/losses.cpython-310.pyc differ
diff --git a/audiocraft/adversarial/discriminators/__pycache__/__init__.cpython-310.pyc b/audiocraft/adversarial/discriminators/__pycache__/__init__.cpython-310.pyc
index 5674f791a892574d891a3a2d42288350fc3aa11f..1ff5852988fd97cd3bbfeafc58e15b3a394efe87 100644
Binary files a/audiocraft/adversarial/discriminators/__pycache__/__init__.cpython-310.pyc and b/audiocraft/adversarial/discriminators/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/adversarial/discriminators/__pycache__/base.cpython-310.pyc b/audiocraft/adversarial/discriminators/__pycache__/base.cpython-310.pyc
index 31d481a5d8f529b968ec5d8350689f7316b3478b..914227680e3e85880711c6e1902ab3a590b34b46 100644
Binary files a/audiocraft/adversarial/discriminators/__pycache__/base.cpython-310.pyc and b/audiocraft/adversarial/discriminators/__pycache__/base.cpython-310.pyc differ
diff --git a/audiocraft/adversarial/discriminators/__pycache__/mpd.cpython-310.pyc b/audiocraft/adversarial/discriminators/__pycache__/mpd.cpython-310.pyc
index 460ab35690ef6662ece4a2d48527c6818cd917e0..d486a8d0a8214f29e9b98a2a97c4732bc990a60a 100644
Binary files a/audiocraft/adversarial/discriminators/__pycache__/mpd.cpython-310.pyc and b/audiocraft/adversarial/discriminators/__pycache__/mpd.cpython-310.pyc differ
diff --git a/audiocraft/adversarial/discriminators/__pycache__/msd.cpython-310.pyc b/audiocraft/adversarial/discriminators/__pycache__/msd.cpython-310.pyc
index e93ade43fc4625d1754ecd84d1d034a8bafb7c68..f3f4f118c9b740c12f8d30b1ddd3604569bdabe0 100644
Binary files a/audiocraft/adversarial/discriminators/__pycache__/msd.cpython-310.pyc and b/audiocraft/adversarial/discriminators/__pycache__/msd.cpython-310.pyc differ
diff --git a/audiocraft/adversarial/discriminators/__pycache__/msstftd.cpython-310.pyc b/audiocraft/adversarial/discriminators/__pycache__/msstftd.cpython-310.pyc
index 37d6f1bdfd3a2b46b56c078ebf0f7d3e039e3100..8195997d45731a9c6fe164f34c2b07eae587115c 100644
Binary files a/audiocraft/adversarial/discriminators/__pycache__/msstftd.cpython-310.pyc and b/audiocraft/adversarial/discriminators/__pycache__/msstftd.cpython-310.pyc differ
diff --git a/audiocraft/data/__pycache__/__init__.cpython-310.pyc b/audiocraft/data/__pycache__/__init__.cpython-310.pyc
index 06bee6fe008abcee3a4cd5b08d7921462255fae8..95c3dcf0c632d2ff8329504ff25f13da8338d69a 100644
Binary files a/audiocraft/data/__pycache__/__init__.cpython-310.pyc and b/audiocraft/data/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/data/__pycache__/audio.cpython-310.pyc b/audiocraft/data/__pycache__/audio.cpython-310.pyc
index 88f43cfd25a4bb5a32f088a1808115a390f6d49c..0379796fd44afea26f9d84a1bbb15755bcf8b94d 100644
Binary files a/audiocraft/data/__pycache__/audio.cpython-310.pyc and b/audiocraft/data/__pycache__/audio.cpython-310.pyc differ
diff --git a/audiocraft/data/__pycache__/audio_dataset.cpython-310.pyc b/audiocraft/data/__pycache__/audio_dataset.cpython-310.pyc
index 1a474c00b076ba98ec794d0f4b41001ea8c32b45..4b91e977a6b8cc3aaf0a49c3fce9f9a90bd3be56 100644
Binary files a/audiocraft/data/__pycache__/audio_dataset.cpython-310.pyc and b/audiocraft/data/__pycache__/audio_dataset.cpython-310.pyc differ
diff --git a/audiocraft/data/__pycache__/audio_utils.cpython-310.pyc b/audiocraft/data/__pycache__/audio_utils.cpython-310.pyc
index 8da8487a7cbda680dbd0d1feef5929dd401eb8ee..3ff56a56f057a17ee4a4892cd8400dff5ffd3ebd 100644
Binary files a/audiocraft/data/__pycache__/audio_utils.cpython-310.pyc and b/audiocraft/data/__pycache__/audio_utils.cpython-310.pyc differ
diff --git a/audiocraft/data/__pycache__/info_audio_dataset.cpython-310.pyc b/audiocraft/data/__pycache__/info_audio_dataset.cpython-310.pyc
index 6fdae11a46e6f69111d518fdf49c3f3d4cb2b0dd..f9d2a5dff57fd31aa79bc723c74af29d6a19e8cb 100644
Binary files a/audiocraft/data/__pycache__/info_audio_dataset.cpython-310.pyc and b/audiocraft/data/__pycache__/info_audio_dataset.cpython-310.pyc differ
diff --git a/audiocraft/data/__pycache__/music_dataset.cpython-310.pyc b/audiocraft/data/__pycache__/music_dataset.cpython-310.pyc
index c1d6739469e3168cc8af78d4c2115252855121dc..4b860c3348f5bcc5d5e331705773f4b053950c44 100644
Binary files a/audiocraft/data/__pycache__/music_dataset.cpython-310.pyc and b/audiocraft/data/__pycache__/music_dataset.cpython-310.pyc differ
diff --git a/audiocraft/data/__pycache__/sound_dataset.cpython-310.pyc b/audiocraft/data/__pycache__/sound_dataset.cpython-310.pyc
index 001dc9c00c2ee0ec4f67827a6e40d9f25c3e7342..c0862fed4772991f7a8f9f4a701b73fb3d447272 100644
Binary files a/audiocraft/data/__pycache__/sound_dataset.cpython-310.pyc and b/audiocraft/data/__pycache__/sound_dataset.cpython-310.pyc differ
diff --git a/audiocraft/data/__pycache__/zip.cpython-310.pyc b/audiocraft/data/__pycache__/zip.cpython-310.pyc
index 13b3b489b97cddad5b00efc0ffa1f519064ff944..62c5ee70fcb94bc0175f4243228997254f82f75f 100644
Binary files a/audiocraft/data/__pycache__/zip.cpython-310.pyc and b/audiocraft/data/__pycache__/zip.cpython-310.pyc differ
diff --git a/audiocraft/losses/__pycache__/__init__.cpython-310.pyc b/audiocraft/losses/__pycache__/__init__.cpython-310.pyc
index c3f96d9484a8bb16e710c6c09cad8c774c0fee4e..c72897a59a50ab3a981f08fa9294dad864f50ca9 100644
Binary files a/audiocraft/losses/__pycache__/__init__.cpython-310.pyc and b/audiocraft/losses/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/losses/__pycache__/balancer.cpython-310.pyc b/audiocraft/losses/__pycache__/balancer.cpython-310.pyc
index ed1268b546ed1eed6dd0ded696e1bdc22a0c3c96..bbdcc19443d16c49a76167ca95a0ecce824ddbd3 100644
Binary files a/audiocraft/losses/__pycache__/balancer.cpython-310.pyc and b/audiocraft/losses/__pycache__/balancer.cpython-310.pyc differ
diff --git a/audiocraft/losses/__pycache__/sisnr.cpython-310.pyc b/audiocraft/losses/__pycache__/sisnr.cpython-310.pyc
index d5a60ac45a81c68651002f7a44b6f1a33284453c..9d471d08cd44514cc02d60b33feb65f7bbc85a07 100644
Binary files a/audiocraft/losses/__pycache__/sisnr.cpython-310.pyc and b/audiocraft/losses/__pycache__/sisnr.cpython-310.pyc differ
diff --git a/audiocraft/losses/__pycache__/specloss.cpython-310.pyc b/audiocraft/losses/__pycache__/specloss.cpython-310.pyc
index 8ebf4b3e2ad137a2c83aacaa993799887c5da8ab..a8e6e61623cc575183eff1e7b1a4bd3346b3c2ef 100644
Binary files a/audiocraft/losses/__pycache__/specloss.cpython-310.pyc and b/audiocraft/losses/__pycache__/specloss.cpython-310.pyc differ
diff --git a/audiocraft/losses/__pycache__/stftloss.cpython-310.pyc b/audiocraft/losses/__pycache__/stftloss.cpython-310.pyc
index b2a0745f846381df970c7d9c09a39b1e34f14b94..09a7ff39487afda8c858ff1ca9d681033526b963 100644
Binary files a/audiocraft/losses/__pycache__/stftloss.cpython-310.pyc and b/audiocraft/losses/__pycache__/stftloss.cpython-310.pyc differ
diff --git a/audiocraft/metrics/__pycache__/__init__.cpython-310.pyc b/audiocraft/metrics/__pycache__/__init__.cpython-310.pyc
index b49fa12b6d6295e40ac59eadfd05566672644be2..075fb392b8010575faddd4c0d5a54612417250de 100644
Binary files a/audiocraft/metrics/__pycache__/__init__.cpython-310.pyc and b/audiocraft/metrics/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/metrics/__pycache__/chroma_cosinesim.cpython-310.pyc b/audiocraft/metrics/__pycache__/chroma_cosinesim.cpython-310.pyc
index 8be3faa7bd05d57b9100a7a43bf78bb2d662046c..a326fca4a97cbdd77a6b02a53c92a1830bb2529b 100644
Binary files a/audiocraft/metrics/__pycache__/chroma_cosinesim.cpython-310.pyc and b/audiocraft/metrics/__pycache__/chroma_cosinesim.cpython-310.pyc differ
diff --git a/audiocraft/metrics/__pycache__/clap_consistency.cpython-310.pyc b/audiocraft/metrics/__pycache__/clap_consistency.cpython-310.pyc
index 0306e9297ff04efdba0e1b5194f35460a3b8a01a..6c234e9e395443a88a03eb2780c4f67289854e0c 100644
Binary files a/audiocraft/metrics/__pycache__/clap_consistency.cpython-310.pyc and b/audiocraft/metrics/__pycache__/clap_consistency.cpython-310.pyc differ
diff --git a/audiocraft/metrics/__pycache__/fad.cpython-310.pyc b/audiocraft/metrics/__pycache__/fad.cpython-310.pyc
index 76407e6c4a0d98bf0d6b30fb7f20f076d62a8e38..f2b6f3d0eb3f2c664f36e81f1b3e07552921259b 100644
Binary files a/audiocraft/metrics/__pycache__/fad.cpython-310.pyc and b/audiocraft/metrics/__pycache__/fad.cpython-310.pyc differ
diff --git a/audiocraft/metrics/__pycache__/kld.cpython-310.pyc b/audiocraft/metrics/__pycache__/kld.cpython-310.pyc
index 832888cea28be1844eb37e978e247717cb7ed223..d0999323b5980f0516f2651685dc7509ff208c34 100644
Binary files a/audiocraft/metrics/__pycache__/kld.cpython-310.pyc and b/audiocraft/metrics/__pycache__/kld.cpython-310.pyc differ
diff --git a/audiocraft/metrics/__pycache__/rvm.cpython-310.pyc b/audiocraft/metrics/__pycache__/rvm.cpython-310.pyc
index 4f3745c7ae024300f04b10bb2a7f8b01809c1fd2..a675e0fd3ac2db85b7fd8e5920e64bf848906b0c 100644
Binary files a/audiocraft/metrics/__pycache__/rvm.cpython-310.pyc and b/audiocraft/metrics/__pycache__/rvm.cpython-310.pyc differ
diff --git a/audiocraft/metrics/__pycache__/visqol.cpython-310.pyc b/audiocraft/metrics/__pycache__/visqol.cpython-310.pyc
index 815e7825c1356117b867456eb92c5295cfb3468b..bb9c55e48455ab18425a865f20c77f775217c271 100644
Binary files a/audiocraft/metrics/__pycache__/visqol.cpython-310.pyc and b/audiocraft/metrics/__pycache__/visqol.cpython-310.pyc differ
diff --git a/audiocraft/models/__pycache__/__init__.cpython-310.pyc b/audiocraft/models/__pycache__/__init__.cpython-310.pyc
index 8026d3a100e97b2b5e96a7a829e857ac6ee21c82..fd4509dbdfb023312f790e1235be833bdc48a207 100644
Binary files a/audiocraft/models/__pycache__/__init__.cpython-310.pyc and b/audiocraft/models/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/models/__pycache__/audiogen.cpython-310.pyc b/audiocraft/models/__pycache__/audiogen.cpython-310.pyc
index 4c7c3096a9371c82be956bf7d82d72d0723cc894..d34297dcae22ff599918e4ec936305ea0606726b 100644
Binary files a/audiocraft/models/__pycache__/audiogen.cpython-310.pyc and b/audiocraft/models/__pycache__/audiogen.cpython-310.pyc differ
diff --git a/audiocraft/models/__pycache__/builders.cpython-310.pyc b/audiocraft/models/__pycache__/builders.cpython-310.pyc
index 5ddca8d54c4bb0518376614e29843125afdd2562..bcedb283bc4a8c1f6eed9c57badf0ef2a89b7497 100644
Binary files a/audiocraft/models/__pycache__/builders.cpython-310.pyc and b/audiocraft/models/__pycache__/builders.cpython-310.pyc differ
diff --git a/audiocraft/models/__pycache__/encodec.cpython-310.pyc b/audiocraft/models/__pycache__/encodec.cpython-310.pyc
index 4eae86f026b93c8f0675b93f899e23bee43ecbb1..729e6c990bb8d3749e5d28b0d1b4846400d5e0d8 100644
Binary files a/audiocraft/models/__pycache__/encodec.cpython-310.pyc and b/audiocraft/models/__pycache__/encodec.cpython-310.pyc differ
diff --git a/audiocraft/models/__pycache__/lm.cpython-310.pyc b/audiocraft/models/__pycache__/lm.cpython-310.pyc
index 8e17a05404a6e67140f74027dd9c39fc00c8b9ca..e087fa938527bbbc721777d171a82ddce7f81395 100644
Binary files a/audiocraft/models/__pycache__/lm.cpython-310.pyc and b/audiocraft/models/__pycache__/lm.cpython-310.pyc differ
diff --git a/audiocraft/models/__pycache__/loaders.cpython-310.pyc b/audiocraft/models/__pycache__/loaders.cpython-310.pyc
index e7c48c9466426d9b309674cd9c306c745cd9199e..29aa0aeb6ce81fd3f7889053b2b1e32e72222827 100644
Binary files a/audiocraft/models/__pycache__/loaders.cpython-310.pyc and b/audiocraft/models/__pycache__/loaders.cpython-310.pyc differ
diff --git a/audiocraft/models/__pycache__/multibanddiffusion.cpython-310.pyc b/audiocraft/models/__pycache__/multibanddiffusion.cpython-310.pyc
index e3ba742e63bd447958b3265f17731aae38ffb26a..f1f481967b10d61dda054c2347aa4d15245651af 100644
Binary files a/audiocraft/models/__pycache__/multibanddiffusion.cpython-310.pyc and b/audiocraft/models/__pycache__/multibanddiffusion.cpython-310.pyc differ
diff --git a/audiocraft/models/__pycache__/musicgen.cpython-310.pyc b/audiocraft/models/__pycache__/musicgen.cpython-310.pyc
index 1ec6de5b70138c4357a33068aeec1c395fd19585..373ab2ac081a0d3fce060106ba19bb1901740eef 100644
Binary files a/audiocraft/models/__pycache__/musicgen.cpython-310.pyc and b/audiocraft/models/__pycache__/musicgen.cpython-310.pyc differ
diff --git a/audiocraft/models/__pycache__/unet.cpython-310.pyc b/audiocraft/models/__pycache__/unet.cpython-310.pyc
index 8fc769e01fd845bfb783c3bb5daf2155630d4963..224ff03d0896e650e19c4b197352a02e828bcc62 100644
Binary files a/audiocraft/models/__pycache__/unet.cpython-310.pyc and b/audiocraft/models/__pycache__/unet.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/__init__.cpython-310.pyc b/audiocraft/modules/__pycache__/__init__.cpython-310.pyc
index 16128c9090c63b33bed7eca6dba0a062fd74e3be..bae03db8c1c771edecb650e94745418a76234fca 100644
Binary files a/audiocraft/modules/__pycache__/__init__.cpython-310.pyc and b/audiocraft/modules/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/activations.cpython-310.pyc b/audiocraft/modules/__pycache__/activations.cpython-310.pyc
index e5559f532116b8b5cbc03544ce73964869552e5f..0070fd8bf621ed526d33d732121006d286e8718f 100644
Binary files a/audiocraft/modules/__pycache__/activations.cpython-310.pyc and b/audiocraft/modules/__pycache__/activations.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/chroma.cpython-310.pyc b/audiocraft/modules/__pycache__/chroma.cpython-310.pyc
index 82a258aaf0454b69810617623db9c771d0a5b672..30ac998cf76566340d0fb01f353de7bca8923691 100644
Binary files a/audiocraft/modules/__pycache__/chroma.cpython-310.pyc and b/audiocraft/modules/__pycache__/chroma.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/codebooks_patterns.cpython-310.pyc b/audiocraft/modules/__pycache__/codebooks_patterns.cpython-310.pyc
index 3a94527edf3980ed2b24ce8b963c96af979aa9d7..9954ac922d3762a2b73237da01084bb779e50af0 100644
Binary files a/audiocraft/modules/__pycache__/codebooks_patterns.cpython-310.pyc and b/audiocraft/modules/__pycache__/codebooks_patterns.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/conditioners.cpython-310.pyc b/audiocraft/modules/__pycache__/conditioners.cpython-310.pyc
index 48617cf2f490744c73471580f58f7195aa842660..4d0805365ad77f7de3361dc54d996d198e2e70ea 100644
Binary files a/audiocraft/modules/__pycache__/conditioners.cpython-310.pyc and b/audiocraft/modules/__pycache__/conditioners.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/conv.cpython-310.pyc b/audiocraft/modules/__pycache__/conv.cpython-310.pyc
index 086b957dcb48fc69edbf87df5e0169a227207165..8b3e0490ab9ddabee1905595f59e9624cc0609e7 100644
Binary files a/audiocraft/modules/__pycache__/conv.cpython-310.pyc and b/audiocraft/modules/__pycache__/conv.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/diffusion_schedule.cpython-310.pyc b/audiocraft/modules/__pycache__/diffusion_schedule.cpython-310.pyc
index c2bb2b600c025ae27e60e42d0adba9712ea69e0c..82cd48f5895a1f1773784386734b1ad4d841aa5a 100644
Binary files a/audiocraft/modules/__pycache__/diffusion_schedule.cpython-310.pyc and b/audiocraft/modules/__pycache__/diffusion_schedule.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/lstm.cpython-310.pyc b/audiocraft/modules/__pycache__/lstm.cpython-310.pyc
index 145810605494378dc35c445fef14df7bc9a50eae..6238a16a15490395175c4b7ce42ea9b1c9bcfe61 100644
Binary files a/audiocraft/modules/__pycache__/lstm.cpython-310.pyc and b/audiocraft/modules/__pycache__/lstm.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/rope.cpython-310.pyc b/audiocraft/modules/__pycache__/rope.cpython-310.pyc
index 5e8bdf73f68e8ffea96e69bb8b3acc4520bb5cfc..9e86a76e573fa6dbc11eebfa04e94b9b00329ed0 100644
Binary files a/audiocraft/modules/__pycache__/rope.cpython-310.pyc and b/audiocraft/modules/__pycache__/rope.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/seanet.cpython-310.pyc b/audiocraft/modules/__pycache__/seanet.cpython-310.pyc
index 278875bcd7bbfc807bc831efe02a647d1863447d..5386679073fe3b7fdcd5eb50d4286a0ea5604770 100644
Binary files a/audiocraft/modules/__pycache__/seanet.cpython-310.pyc and b/audiocraft/modules/__pycache__/seanet.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/streaming.cpython-310.pyc b/audiocraft/modules/__pycache__/streaming.cpython-310.pyc
index d3f6f33575ae4ee37d5952ae8081c2194f8c0fa4..10b56033c3472023c2eca7c124b0a30ddd8696ee 100644
Binary files a/audiocraft/modules/__pycache__/streaming.cpython-310.pyc and b/audiocraft/modules/__pycache__/streaming.cpython-310.pyc differ
diff --git a/audiocraft/modules/__pycache__/transformer.cpython-310.pyc b/audiocraft/modules/__pycache__/transformer.cpython-310.pyc
index 64dbfc5130bb90d19d409671baa4ae7d1cee2d6a..4d549972f3f73a4ed46defe9ec8147ed88502243 100644
Binary files a/audiocraft/modules/__pycache__/transformer.cpython-310.pyc and b/audiocraft/modules/__pycache__/transformer.cpython-310.pyc differ
diff --git a/audiocraft/optim/__pycache__/__init__.cpython-310.pyc b/audiocraft/optim/__pycache__/__init__.cpython-310.pyc
index 6a8fa09fc77f29ff6c89e162cadee7f240b14a13..09259e5cb10fba1b5112cd3c14099f11ff89fa3a 100644
Binary files a/audiocraft/optim/__pycache__/__init__.cpython-310.pyc and b/audiocraft/optim/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/optim/__pycache__/cosine_lr_scheduler.cpython-310.pyc b/audiocraft/optim/__pycache__/cosine_lr_scheduler.cpython-310.pyc
index e51656014e1f8d36f7ba249527b04827d186e9f8..15dd337e12a826130a18952a2ba22cc2779e846b 100644
Binary files a/audiocraft/optim/__pycache__/cosine_lr_scheduler.cpython-310.pyc and b/audiocraft/optim/__pycache__/cosine_lr_scheduler.cpython-310.pyc differ
diff --git a/audiocraft/optim/__pycache__/dadam.cpython-310.pyc b/audiocraft/optim/__pycache__/dadam.cpython-310.pyc
index 48bd9b287a9cfb1958f980f2b77f1b497a3dc1dc..10ad74b9296472023d3a30dcf1fb8415a7e16190 100644
Binary files a/audiocraft/optim/__pycache__/dadam.cpython-310.pyc and b/audiocraft/optim/__pycache__/dadam.cpython-310.pyc differ
diff --git a/audiocraft/optim/__pycache__/ema.cpython-310.pyc b/audiocraft/optim/__pycache__/ema.cpython-310.pyc
index cc3f7601700f72d9f1f7d552b9022c2c9030183c..926892c6fc231772c5c5851e30b84973c96b8c59 100644
Binary files a/audiocraft/optim/__pycache__/ema.cpython-310.pyc and b/audiocraft/optim/__pycache__/ema.cpython-310.pyc differ
diff --git a/audiocraft/optim/__pycache__/fsdp.cpython-310.pyc b/audiocraft/optim/__pycache__/fsdp.cpython-310.pyc
index ae2738e2f44890001cea399bb1955a48e0409211..7303768fd2a8ae213fe68d7ee11255db32fbce52 100644
Binary files a/audiocraft/optim/__pycache__/fsdp.cpython-310.pyc and b/audiocraft/optim/__pycache__/fsdp.cpython-310.pyc differ
diff --git a/audiocraft/optim/__pycache__/inverse_sqrt_lr_scheduler.cpython-310.pyc b/audiocraft/optim/__pycache__/inverse_sqrt_lr_scheduler.cpython-310.pyc
index 184bfe00927b8e9aecbf587eb11f48287b4c0f5a..a5476a570510f876622a8b56ad2c2b72415387a1 100644
Binary files a/audiocraft/optim/__pycache__/inverse_sqrt_lr_scheduler.cpython-310.pyc and b/audiocraft/optim/__pycache__/inverse_sqrt_lr_scheduler.cpython-310.pyc differ
diff --git a/audiocraft/optim/__pycache__/linear_warmup_lr_scheduler.cpython-310.pyc b/audiocraft/optim/__pycache__/linear_warmup_lr_scheduler.cpython-310.pyc
index 4854a32da7d1de62b30e4aef33136a0f695e788b..69da23a9457cc961ecd08d9ff94e8426d115ea1d 100644
Binary files a/audiocraft/optim/__pycache__/linear_warmup_lr_scheduler.cpython-310.pyc and b/audiocraft/optim/__pycache__/linear_warmup_lr_scheduler.cpython-310.pyc differ
diff --git a/audiocraft/optim/__pycache__/polynomial_decay_lr_scheduler.cpython-310.pyc b/audiocraft/optim/__pycache__/polynomial_decay_lr_scheduler.cpython-310.pyc
index 9d7e9f2e3020fec85d99725f47c24bcecf210347..149224b9f647e707b47caa0ac8d30f0caa541d7c 100644
Binary files a/audiocraft/optim/__pycache__/polynomial_decay_lr_scheduler.cpython-310.pyc and b/audiocraft/optim/__pycache__/polynomial_decay_lr_scheduler.cpython-310.pyc differ
diff --git a/audiocraft/quantization/__pycache__/__init__.cpython-310.pyc b/audiocraft/quantization/__pycache__/__init__.cpython-310.pyc
index f649051a17b492e4b558d31a65fe1738f28eb37f..f027dbc406a4c54efe1835c7da0cd33c1fb266c9 100644
Binary files a/audiocraft/quantization/__pycache__/__init__.cpython-310.pyc and b/audiocraft/quantization/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/quantization/__pycache__/base.cpython-310.pyc b/audiocraft/quantization/__pycache__/base.cpython-310.pyc
index 3c9f10ee3741cc4f4d02ee026d37ec4e1f8009ab..ccee4b1a359086c58da155e71f9a9e0cba28db58 100644
Binary files a/audiocraft/quantization/__pycache__/base.cpython-310.pyc and b/audiocraft/quantization/__pycache__/base.cpython-310.pyc differ
diff --git a/audiocraft/quantization/__pycache__/core_vq.cpython-310.pyc b/audiocraft/quantization/__pycache__/core_vq.cpython-310.pyc
index e7967963440d6c92dcb4ef38bc5cfaa57ddbbcd8..0005af4196380e76b4c331a42e92095362d0c8f0 100644
Binary files a/audiocraft/quantization/__pycache__/core_vq.cpython-310.pyc and b/audiocraft/quantization/__pycache__/core_vq.cpython-310.pyc differ
diff --git a/audiocraft/quantization/__pycache__/vq.cpython-310.pyc b/audiocraft/quantization/__pycache__/vq.cpython-310.pyc
index 68425f7de959173edc618cc44789ee5d29cecbc5..111b6950dfc103c7d4aceaefe158139568fd4488 100644
Binary files a/audiocraft/quantization/__pycache__/vq.cpython-310.pyc and b/audiocraft/quantization/__pycache__/vq.cpython-310.pyc differ
diff --git a/audiocraft/solvers/__pycache__/__init__.cpython-310.pyc b/audiocraft/solvers/__pycache__/__init__.cpython-310.pyc
index e203e6270d827c17050af24e5cb98687478c9713..1b17942a69ae1a24a85b2669c5d12ae98ae7dbf5 100644
Binary files a/audiocraft/solvers/__pycache__/__init__.cpython-310.pyc and b/audiocraft/solvers/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/solvers/__pycache__/audiogen.cpython-310.pyc b/audiocraft/solvers/__pycache__/audiogen.cpython-310.pyc
index 37ac9e25b5f35d849e9a5418dfecc6e78834c813..744d784f7b687332716db6525aecb0a34e4bfe7a 100644
Binary files a/audiocraft/solvers/__pycache__/audiogen.cpython-310.pyc and b/audiocraft/solvers/__pycache__/audiogen.cpython-310.pyc differ
diff --git a/audiocraft/solvers/__pycache__/base.cpython-310.pyc b/audiocraft/solvers/__pycache__/base.cpython-310.pyc
index 0e029719d3a11a74d71343904ec1ea00252ed79f..2f4493446528f6f3c83d903224861354a08fcb0b 100644
Binary files a/audiocraft/solvers/__pycache__/base.cpython-310.pyc and b/audiocraft/solvers/__pycache__/base.cpython-310.pyc differ
diff --git a/audiocraft/solvers/__pycache__/builders.cpython-310.pyc b/audiocraft/solvers/__pycache__/builders.cpython-310.pyc
index 56185350e8455eb9483574b8eec3e51d9dadd88d..9b72e11542d07ea1dc578236678f08c482c67c0b 100644
Binary files a/audiocraft/solvers/__pycache__/builders.cpython-310.pyc and b/audiocraft/solvers/__pycache__/builders.cpython-310.pyc differ
diff --git a/audiocraft/solvers/__pycache__/compression.cpython-310.pyc b/audiocraft/solvers/__pycache__/compression.cpython-310.pyc
index 4d44696a5bddbac5c4690afc0ab795b21fabbd09..87809cd35b51a7d1dd51d194518a5a2460bd991d 100644
Binary files a/audiocraft/solvers/__pycache__/compression.cpython-310.pyc and b/audiocraft/solvers/__pycache__/compression.cpython-310.pyc differ
diff --git a/audiocraft/solvers/__pycache__/diffusion.cpython-310.pyc b/audiocraft/solvers/__pycache__/diffusion.cpython-310.pyc
index 91983e732777e62c6bcf90623a85b7eee1499fe6..586529f9c1aec711fb5184c997edd867647257e4 100644
Binary files a/audiocraft/solvers/__pycache__/diffusion.cpython-310.pyc and b/audiocraft/solvers/__pycache__/diffusion.cpython-310.pyc differ
diff --git a/audiocraft/solvers/__pycache__/musicgen.cpython-310.pyc b/audiocraft/solvers/__pycache__/musicgen.cpython-310.pyc
index de0840f985b2d5f5685d7cf03fba950781437ee8..2a01e90ad592fe0e1845341d7dcb499c3fe099d8 100644
Binary files a/audiocraft/solvers/__pycache__/musicgen.cpython-310.pyc and b/audiocraft/solvers/__pycache__/musicgen.cpython-310.pyc differ
diff --git a/audiocraft/utils/__pycache__/__init__.cpython-310.pyc b/audiocraft/utils/__pycache__/__init__.cpython-310.pyc
index c2fdf8739bcfa1d4d1e0a442529b6513a3ca165d..824453ef19e5e12fbeae393f92c74d56479ddc52 100644
Binary files a/audiocraft/utils/__pycache__/__init__.cpython-310.pyc and b/audiocraft/utils/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/utils/__pycache__/autocast.cpython-310.pyc b/audiocraft/utils/__pycache__/autocast.cpython-310.pyc
index ab01a2a0693e3d7dcd1fd8a5af32dd09c9bddfbc..5e3c72b657d24b5c19c4771557f05ef2befecc86 100644
Binary files a/audiocraft/utils/__pycache__/autocast.cpython-310.pyc and b/audiocraft/utils/__pycache__/autocast.cpython-310.pyc differ
diff --git a/audiocraft/utils/__pycache__/best_state.cpython-310.pyc b/audiocraft/utils/__pycache__/best_state.cpython-310.pyc
index 82c73f7d83af2158e3894987eb334ac8c02aac60..7f3861c864c1bf56f17e3bea74f82647d42e7855 100644
Binary files a/audiocraft/utils/__pycache__/best_state.cpython-310.pyc and b/audiocraft/utils/__pycache__/best_state.cpython-310.pyc differ
diff --git a/audiocraft/utils/__pycache__/cache.cpython-310.pyc b/audiocraft/utils/__pycache__/cache.cpython-310.pyc
index 1c76bbdfb0b94c2d8c9db263b3da298174610293..e30b1d4f3e5ecb2630a3182873904a799c3cba9f 100644
Binary files a/audiocraft/utils/__pycache__/cache.cpython-310.pyc and b/audiocraft/utils/__pycache__/cache.cpython-310.pyc differ
diff --git a/audiocraft/utils/__pycache__/checkpoint.cpython-310.pyc b/audiocraft/utils/__pycache__/checkpoint.cpython-310.pyc
index e9897c55611ff26e6ea37f80a822dabb3528fc55..4921e8b7de2ae6f40b5afa82946701b1a49033da 100644
Binary files a/audiocraft/utils/__pycache__/checkpoint.cpython-310.pyc and b/audiocraft/utils/__pycache__/checkpoint.cpython-310.pyc differ
diff --git a/audiocraft/utils/__pycache__/cluster.cpython-310.pyc b/audiocraft/utils/__pycache__/cluster.cpython-310.pyc
index 44dc18deeac308d0d0213e1750a3e4399ff3835a..fde424670a998c4d8b0f47b94866f26f557660a7 100644
Binary files a/audiocraft/utils/__pycache__/cluster.cpython-310.pyc and b/audiocraft/utils/__pycache__/cluster.cpython-310.pyc differ
diff --git a/audiocraft/utils/__pycache__/deadlock.cpython-310.pyc b/audiocraft/utils/__pycache__/deadlock.cpython-310.pyc
index 3ff3d1d194727b54dd73691fdd8df691ea10f740..505c7f96dd5c8cfb33e87811a336851f187029a9 100644
Binary files a/audiocraft/utils/__pycache__/deadlock.cpython-310.pyc and b/audiocraft/utils/__pycache__/deadlock.cpython-310.pyc differ
diff --git a/audiocraft/utils/__pycache__/profiler.cpython-310.pyc b/audiocraft/utils/__pycache__/profiler.cpython-310.pyc
index 25947b53c6783c5166249a03821990a30f0943fd..6ac1e5e09c9c206bdfa40d4357382dc0826fbcd9 100644
Binary files a/audiocraft/utils/__pycache__/profiler.cpython-310.pyc and b/audiocraft/utils/__pycache__/profiler.cpython-310.pyc differ
diff --git a/audiocraft/utils/__pycache__/ui.cpython-310.pyc b/audiocraft/utils/__pycache__/ui.cpython-310.pyc
index afea185c9fca2558ffecfc3ea9f9ca905da6717e..5a6c22e510dec4a37af9cffd4a37d70c6f1390f5 100644
Binary files a/audiocraft/utils/__pycache__/ui.cpython-310.pyc and b/audiocraft/utils/__pycache__/ui.cpython-310.pyc differ
diff --git a/audiocraft/utils/__pycache__/utils.cpython-310.pyc b/audiocraft/utils/__pycache__/utils.cpython-310.pyc
index 6a261ed941ec215bfa209d8809597a1bae80e9b8..7d77606ab87746016f33e669861d72d6d1bb38fb 100644
Binary files a/audiocraft/utils/__pycache__/utils.cpython-310.pyc and b/audiocraft/utils/__pycache__/utils.cpython-310.pyc differ
diff --git a/audiocraft/utils/samples/__pycache__/__init__.cpython-310.pyc b/audiocraft/utils/samples/__pycache__/__init__.cpython-310.pyc
index b420f4a97da11c1050dab1f6387d5551f4027274..bf984c34e675c96fe779f549b898c5e91ef6abec 100644
Binary files a/audiocraft/utils/samples/__pycache__/__init__.cpython-310.pyc and b/audiocraft/utils/samples/__pycache__/__init__.cpython-310.pyc differ
diff --git a/audiocraft/utils/samples/__pycache__/manager.cpython-310.pyc b/audiocraft/utils/samples/__pycache__/manager.cpython-310.pyc
index f5b35ad585599c5302f6dea9dcd0fe5442314d18..17338bb1f0cd3fa4851ffa641ed74ab81341e46b 100644
Binary files a/audiocraft/utils/samples/__pycache__/manager.cpython-310.pyc and b/audiocraft/utils/samples/__pycache__/manager.cpython-310.pyc differ