diff --git a/audiocraft/__pycache__/__init__.cpython-310.pyc b/audiocraft/__pycache__/__init__.cpython-310.pyc
index 895ed9ab3a31e060453dfaca40434f517c9ea9ae..00aa091cf9657a81fc69af01c51e6e301875dcaa 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 f41ed26a7a3f593f327d329f15b348cef3142512..650017506b36dfef531b5a22c49edffdbdfbbc6d 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 97989ca816af970a2e8b937358c56b87a8577dd0..9734541921143de865d1cb7d8983582a9b3e265c 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 301211b70e061429a84fef7265b97a63992965b4..d6993709432769d60bf5bbe8086d930ae39d2d1b 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 1ff5852988fd97cd3bbfeafc58e15b3a394efe87..a013078df5e56040290962fcbe633afeeb255f44 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 914227680e3e85880711c6e1902ab3a590b34b46..2e8b10755bece7fcf643c3a0edea9a04bf317130 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 d486a8d0a8214f29e9b98a2a97c4732bc990a60a..8b5ade2d427ff8ff3afa0b9eea500f58c63127c2 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 f3f4f118c9b740c12f8d30b1ddd3604569bdabe0..7e15dae05ca396bb3cb32bd28dadd71373f4d117 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 8195997d45731a9c6fe164f34c2b07eae587115c..bd4baf6be2c596e7f0f4e8e1cf6daf35127c1992 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 95c3dcf0c632d2ff8329504ff25f13da8338d69a..9c97932dc2135dbc58c06e036835040ec5ccc2db 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 0379796fd44afea26f9d84a1bbb15755bcf8b94d..0058532f207511aa32f8f88c5023ab9d7746e9a7 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 4b91e977a6b8cc3aaf0a49c3fce9f9a90bd3be56..c7a5a8dbe41686139502ec4a7aa47657c060c66d 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 3ff56a56f057a17ee4a4892cd8400dff5ffd3ebd..955ff6b6e07adf30573516f710a287d47d6d9308 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 f9d2a5dff57fd31aa79bc723c74af29d6a19e8cb..88809964a72eda283ac0bf92ef9c9de1fba70ac1 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 4b860c3348f5bcc5d5e331705773f4b053950c44..ab8d4d0697cbbd1bb09c35949bcc24c2bbc80aac 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 c0862fed4772991f7a8f9f4a701b73fb3d447272..4abffcb193d64b68ae14c7497f2dc8555a972af9 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 62c5ee70fcb94bc0175f4243228997254f82f75f..142dc7e4d55f82f1d597f7863754bb2e4dc718f1 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 c72897a59a50ab3a981f08fa9294dad864f50ca9..a0847d4a985515b9e3d00982c1733846e8ce7ce1 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 bbdcc19443d16c49a76167ca95a0ecce824ddbd3..a256418b2fc431615fe4e27af2ca4ec2e3040657 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 9d471d08cd44514cc02d60b33feb65f7bbc85a07..3949a582c019cfc08f07727acd242d150636912c 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 a8e6e61623cc575183eff1e7b1a4bd3346b3c2ef..ee1c388294a009c8735b0d2ae4fcd5b9ebd0081e 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 09a7ff39487afda8c858ff1ca9d681033526b963..f491e2c9ea3bc0d342e0d3a445eb408dc12cb4ed 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 075fb392b8010575faddd4c0d5a54612417250de..3bda959f124cb6b97589e4a0e55ece1ea82eccd2 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 a326fca4a97cbdd77a6b02a53c92a1830bb2529b..7994e9a80799a5ec94e66225888c38a58906306d 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 6c234e9e395443a88a03eb2780c4f67289854e0c..5481a7e454e7f499022ea789a49515cd257aa87e 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 f2b6f3d0eb3f2c664f36e81f1b3e07552921259b..96ba1fdaeca31459a757e198fa79a2436c594163 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 d0999323b5980f0516f2651685dc7509ff208c34..78fd98a49f8c4e8a6895bfb257e44e73204cbcdb 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 a675e0fd3ac2db85b7fd8e5920e64bf848906b0c..2da4ea19127431882203367b8e219ff64e7be057 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 bb9c55e48455ab18425a865f20c77f775217c271..ddc66d9d2289c4ed287adb296dd837a6067ab9fb 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 fd4509dbdfb023312f790e1235be833bdc48a207..fb5f3918a3d48cc2168b6f2380a24c1d2aa5524b 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 d34297dcae22ff599918e4ec936305ea0606726b..53c2aafdab6870e910b39ed770df46fef0cca3d9 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 bcedb283bc4a8c1f6eed9c57badf0ef2a89b7497..a85c6d75e16c01bf3ed73bab5cf20db69c4828d8 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 729e6c990bb8d3749e5d28b0d1b4846400d5e0d8..a068529b834dd559e29eef5f6ead5f162032228b 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 e087fa938527bbbc721777d171a82ddce7f81395..89892fd3ac56f11da9620478a687953b2b96289f 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 29aa0aeb6ce81fd3f7889053b2b1e32e72222827..93432139fc60337fc742b27a9f2b51d7f6ee6791 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 f1f481967b10d61dda054c2347aa4d15245651af..1fdfe52c6139315f95cd76b76e99f447477b9dcc 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 373ab2ac081a0d3fce060106ba19bb1901740eef..ad4c98cdd539c79dbd90fd1d9d22f8f0924a0aef 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 224ff03d0896e650e19c4b197352a02e828bcc62..435e696381a83ae2322118d82b22859166b56b74 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 bae03db8c1c771edecb650e94745418a76234fca..1d955421216573ecb7bbc7fc0b20afd6f022f3fc 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 0070fd8bf621ed526d33d732121006d286e8718f..2f0e29d6ce662ca14b1b26c4df1ccca264280628 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 30ac998cf76566340d0fb01f353de7bca8923691..2398aa5fe84ceb45ae5c92de15d5b99ccb34c4e5 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 9954ac922d3762a2b73237da01084bb779e50af0..34dda389b1d3f1bed1ba36ea1bde09933b810fe9 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 4d0805365ad77f7de3361dc54d996d198e2e70ea..a31741e8be96fd93c3b25561e83d18d744d0820a 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 8b3e0490ab9ddabee1905595f59e9624cc0609e7..1588ebfe93cd0b90f91e2f9e5e9b259e224cacc3 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 82cd48f5895a1f1773784386734b1ad4d841aa5a..a627e21b2be2c6341f6420d5e48b5598712fc3be 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 6238a16a15490395175c4b7ce42ea9b1c9bcfe61..a94d6788a2fd7f5ccda6464e8bbe76e44b91b543 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 9e86a76e573fa6dbc11eebfa04e94b9b00329ed0..27b8c2606133e3006584de955aa7ce2fc3e636d0 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 5386679073fe3b7fdcd5eb50d4286a0ea5604770..33112fda88a6d3d726cb06dbc7789c8f50424ef9 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 10b56033c3472023c2eca7c124b0a30ddd8696ee..da65f14534093920b609a02ffe3c513ef11ecef1 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 4d549972f3f73a4ed46defe9ec8147ed88502243..a4bc005a306d4f82a9b47eaf4a27c9c4985a28f1 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 09259e5cb10fba1b5112cd3c14099f11ff89fa3a..0b12f1f820385503e0a6f5b746780d7c4246e575 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 15dd337e12a826130a18952a2ba22cc2779e846b..7fa9505174940b6a70d3e3eb0d42996d7d6b4056 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 10ad74b9296472023d3a30dcf1fb8415a7e16190..aa002937e25317df48744e14f425f711c33329fb 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 926892c6fc231772c5c5851e30b84973c96b8c59..15318070eb47c81d08e9ab64e77712383ed61d23 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 7303768fd2a8ae213fe68d7ee11255db32fbce52..5a630d6b5bc982f7d5f230c39c7b9db977cd31d2 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 a5476a570510f876622a8b56ad2c2b72415387a1..206347e031e5b4f6e4ef80582dc2bbdb443f45ff 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 69da23a9457cc961ecd08d9ff94e8426d115ea1d..54163e4f9685a965dbac65b114053ae24b5be114 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 149224b9f647e707b47caa0ac8d30f0caa541d7c..88f479f25a25bf8d5de88035abfbc26b72b5eedd 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 f027dbc406a4c54efe1835c7da0cd33c1fb266c9..97e760136eda7003ba44a56c507322e6a7cd8326 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 ccee4b1a359086c58da155e71f9a9e0cba28db58..d2fdd02546808b6e4161fda5beb28e93a344a21d 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 0005af4196380e76b4c331a42e92095362d0c8f0..c7f6bfe9f4b47ba1133438e2f450f1dfeff5a88a 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 111b6950dfc103c7d4aceaefe158139568fd4488..336bdc94684e890815d4eacd1eba2885f9307a69 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 1b17942a69ae1a24a85b2669c5d12ae98ae7dbf5..3276de1570d492bd7599b418c8a9310c6852d754 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 744d784f7b687332716db6525aecb0a34e4bfe7a..1fdc8648e32394cfeea4811e9bd433915b83f913 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 2f4493446528f6f3c83d903224861354a08fcb0b..b11efead0bfe479ef19de6f4afac842fb776e75d 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 9b72e11542d07ea1dc578236678f08c482c67c0b..694094a24083dd4c9ce06dac515f5ecbdeada6ce 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 87809cd35b51a7d1dd51d194518a5a2460bd991d..839d184d8837b5813967840820c6de9c1d5df2cc 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 586529f9c1aec711fb5184c997edd867647257e4..625db3e91559961658e594b26e28f3ad231e1721 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 2a01e90ad592fe0e1845341d7dcb499c3fe099d8..c8b3e777a3a4278daaa71d1f8c0100a262acc7c2 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 824453ef19e5e12fbeae393f92c74d56479ddc52..fe2761d9f0139827b5e1127f60f0a31093f053fb 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 5e3c72b657d24b5c19c4771557f05ef2befecc86..011c6203566e9e570ae9401dcd4f181868aa4b42 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 7f3861c864c1bf56f17e3bea74f82647d42e7855..bf86b44054733d26982153bbd6294b7ff17297aa 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 e30b1d4f3e5ecb2630a3182873904a799c3cba9f..42eb77df9e15c0f71c13b91a6378bd9ba6304e8c 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 4921e8b7de2ae6f40b5afa82946701b1a49033da..16e6bf983e21ed7097d0a74971f80f6a5079dbde 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 fde424670a998c4d8b0f47b94866f26f557660a7..ccf727fcb3b656e46d2e6ab6e2e76aeaf934df5f 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 505c7f96dd5c8cfb33e87811a336851f187029a9..c610901de215b195854f9805a1d4a968badf34e6 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 6ac1e5e09c9c206bdfa40d4357382dc0826fbcd9..9a9af9367add1f700bb39a3423d076cadda9c71e 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 5a6c22e510dec4a37af9cffd4a37d70c6f1390f5..9f3993b742b8667b9990fd50dcede012106220ae 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 7d77606ab87746016f33e669861d72d6d1bb38fb..2c32320d9880ed51b06fd3c7c1cb4670e7bdf78e 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 bf984c34e675c96fe779f549b898c5e91ef6abec..c2ffd0ffdc9bb9cc7f60a5c5b3efd323e7d42d16 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 17338bb1f0cd3fa4851ffa641ed74ab81341e46b..43e983b574043fe2ffea8d6848e00227dcc3f1aa 100644
Binary files a/audiocraft/utils/samples/__pycache__/manager.cpython-310.pyc and b/audiocraft/utils/samples/__pycache__/manager.cpython-310.pyc differ