From 51e9e14673b56dd807134f3ca1dd236036cc1a55 Mon Sep 17 00:00:00 2001 From: Gregory Comer Date: Tue, 18 Nov 2025 16:37:23 -0800 Subject: [PATCH] Manually pick moshi requirements update --- examples/models/moshi/mimi/install_requirements.sh | 8 ++++---- examples/models/moshi/mimi/test_mimi.py | 3 +-- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/examples/models/moshi/mimi/install_requirements.sh b/examples/models/moshi/mimi/install_requirements.sh index cfe691c7bd4..20273f5fdac 100755 --- a/examples/models/moshi/mimi/install_requirements.sh +++ b/examples/models/moshi/mimi/install_requirements.sh @@ -7,10 +7,10 @@ set -x -conda install -c conda-forge "ffmpeg<8" -y -pip install torchcodec==0.7.0.dev20250906 --extra-index-url https://download.pytorch.org/whl/nightly/cpu -pip install moshi==0.2.4 -pip install bitsandbytes soundfile +sudo apt install ffmpeg -y +pip install torchcodec==0.7.0.dev20251012 --extra-index-url https://download.pytorch.org/whl/nightly/cpu +pip install moshi==0.2.11 +pip install bitsandbytes soundfile einops # Run llama2/install requirements for torchao deps SCRIPT_DIR=$( cd -- "$( dirname -- "${BASH_SOURCE[0]}" )" &> /dev/null && pwd ) bash "$SCRIPT_DIR"/../../llama/install_requirements.sh diff --git a/examples/models/moshi/mimi/test_mimi.py b/examples/models/moshi/mimi/test_mimi.py index be3c075913d..915c52effbe 100644 --- a/examples/models/moshi/mimi/test_mimi.py +++ b/examples/models/moshi/mimi/test_mimi.py @@ -189,8 +189,7 @@ def forward(self, x): x = self.mimi_model.upsample(x) (emb,) = self.mimi_model.decoder_transformer(x) emb.transpose(1, 2) - with self.mimi_model._context_for_encoder_decoder: - out = self.mimi_model.decoder(emb) + out = self.mimi_model.decoder(emb) return out emb_input = torch.rand(1, 1, 512, device="cpu")