PiCoGen / requirements.txt
Vansh Chugh
vendor jukebox+sheetsage source and recovered weights, use_jukebox=True
ff7b988
Raw
History Blame Contribute Delete
614 Bytes
git+https://github.com/TEAMuP-dev/pyharp.git@v0.3.0
# model-specific deps below:
setuptools<82 # madmom imports pkg_resources at runtime; removed in setuptools 82+
torch
torchaudio
transformers
miditoolkit
questionary
soundfile
mpi4py
beat_this @ https://github.com/CPJKU/beat_this/archive/main.zip
madmom @ git+https://github.com/CPJKU/madmom.git@0551aa8f48d71a367d92b5d3a347a0cf7cd97cc9
# sheetsage/ and jukebox/ are vendored directly via this repo since jukebox's own
# packaging is broken upstream. Here are their runtime deps:
numpy<2
scipy
pretty-midi
validators
pillow
fire
unidecode
numba
librosa
resampy