diff --git a/moshi/moshi/server.py b/moshi/moshi/server.py index 771f491d..a2e596d5 100644 --- a/moshi/moshi/server.py +++ b/moshi/moshi/server.py @@ -426,7 +426,9 @@ def main(): # Download config.json to increment download counter # No worries about double-counting since config.json will be cached the second time - hf_hub_download(args.hf_repo, "config.json") + # Skip when running in offline mode (HF_HUB_OFFLINE=1) + if not os.environ.get("HF_HUB_OFFLINE"): + hf_hub_download(args.hf_repo, "config.json") logger.info("loading mimi") if args.mimi_weight is None: diff --git a/moshi/pyproject.toml b/moshi/pyproject.toml index ead71e9c..bd20d1a5 100644 --- a/moshi/pyproject.toml +++ b/moshi/pyproject.toml @@ -10,7 +10,7 @@ dependencies = [ "sentencepiece == 0.2", "sounddevice == 0.5", "sphn >= 0.1.4, < 0.2", - "torch >= 2.2.0, < 2.5", + "torch >= 2.2.0", "aiohttp>=3.10.5, <3.11", ] authors = [{name="Rajarshi Roy", email="rajarshir@nvidia.com"}]