Looper

andreasjansson · Jun 18, 2023 · 2f80031 · 2f80031
1 parent 401d6db
commit 2f80031
Show file tree

Hide file tree

Showing 3 changed files with 275 additions and 402 deletions.
diff --git a/cog.yaml b/cog.yaml
@@ -7,22 +7,22 @@ build:
   cuda: "11.7"
 
   # a list of ubuntu apt packages to install
-  # system_packages:
-    # - "libgl1-mesa-glx"
-    # - "libglib2.0-0"
+  system_packages:
+    - "ffmpeg"
+    - "portaudio19-dev"
+    - "rubberband-cli"
 
   # python version in the form '3.8' or '3.8.12'
   python_version: "3.9"
 
   # a list of packages in the format <package-name>==<version>
   python_packages:
     - "torch==2.0.1"
-
-  # commands run after the environment is setup
-  run:
-    - "pip install git+https://github.com/facebookresearch/audiocraft.git"
-    - "apt-get update && apt-get install -y ffmpeg"
+    - "git+https://github.com/facebookresearch/audiocraft.git"
+    - "pyaudio==0.2.13"
+    - "git+https://github.com/CPJKU/madmom"
+    - "BeatNet==1.1.0"
+    - "pyrubberband==0.3.0"
 
 # predict.py defines how predictions are run on your model
 predict: "predict.py:Predictor"
-train: "train.py:train"