loopgen / model.json
Vansh Chugh
initial deploy
83e59db
Raw
History Blame Contribute Delete
2.13 kB
{
"name": "loopgen",
"package_dir": "src",
"entry_point": ["src.loopgen.LoopGen", "src.audiocraft.models.musicgen.MusicGen"],
"notes": "Multi-package model. (1) src/loopgen.py + src/lm_loopgen.py are LoopGen's own MAGNeT wrapper, copied from loopgen-repo/src/. (2) src/audiocraft/ is audiocraft v1.3.0 (loopgen-repo's own pin) vendored from source rather than pip-installed, because PyPI audiocraft==1.3.0 hard-pins torch==2.1.0/av==11.0.0/xformers<0.0.23, which conflicts with the torch 2.8.0 ZeroGPU needs -- vendoring sidesteps that pin entirely (same pattern already used by teamup-tech/instruct-MusicGen). All internal absolute 'from audiocraft.X' imports were mechanically rewritten to 'from src.audiocraft.X'; relative imports untouched. models/__init__.py and data/__init__.py were trimmed to drop AudioGen/MultiBandDiffusion/DiffusionUnet/DAC and training-dataset classes (unused, and multibanddiffusion pulls in the training-only solvers package). loaders.py's three torch.load calls got weights_only=False (MAGNeT's checkpoint bundles an OmegaConf config, not just tensors). (3) beat_this/ is vendored separately from ~/Documents/GitHub/beat-this (NOT loopgen's own git+ pin of CPJKU/beat_this) -- same upstream model, already proven on torch 2.11.0 via teamup-tech's own beat-this Space. See SOURCES.md.",
"additional_vendored_packages": [
{
"package_dir": "beat_this",
"entry_point": "beat_this.inference.File2Beats",
"vendored_from": "~/Documents/GitHub/beat-this/beat_this/ (local clone, not loopgen-repo)"
}
],
"checkpoints": {
"musicgen-medium": "auto-downloaded from facebook/musicgen-medium via audiocraft's MusicGen.get_pretrained() -- no manual hosting",
"magnet-medium-30secs": "auto-downloaded from facebook/magnet-medium-30secs via audiocraft's LoopGen.get_pretrained() -- no manual hosting",
"beat_this final0.ckpt": {"repo": "teamup-tech/beat-this", "repo_type": "space", "filename": "final0.ckpt", "size_mb": 81, "note": "reused from the already-deployed teamup-tech/beat-this Space via hf_hub_download(repo_type=\"space\"), not re-uploaded"}
}
}