Init commit

.gitignore

@@ -0,0 +1,6 @@
+.ipynb_checkpoints/
+env/
+wandb/
+*.egg-info/
+data/
+.DS_Store

main.py

@@ -0,0 +1,19 @@
+from audio_diffusion_pytorch import AudioDiffusionModel
+import torch
+from tqdm import tqdm
+import wandb
+
+model = AudioDiffusionModel(in_channels=1)
+wandb.init(project="RemFX", entity="mattricesound")
+
+x = torch.randn(2, 1, 2**18)
+for i in tqdm(range(100)):
+    loss = model(x)
+    loss.backward()
+    if i % 10 == 0:
+        print(loss)
+        wandb.log({"loss": loss})
+
+
+noise = torch.randn(2, 1, 2**18)
+sampled = model.sample(noise=noise, num_steps=5)

setup.py

@@ -0,0 +1,49 @@
+from pathlib import Path
+from setuptools import setup, find_packages
+
+NAME = "REMFX"
+DESCRIPTION = ""
+URL = ""
+EMAIL = "[email protected]"
+AUTHOR = "Matthew Rice"
+REQUIRES_PYTHON = ">=3.8.0"
+VERSION = "0.0.1"
+
+HERE = Path(__file__).parent
+
+try:
+    with open(HERE / "README.md", encoding="utf-8") as f:
+        long_description = "\n" + f.read()
+except FileNotFoundError:
+    long_description = DESCRIPTION
+
+setup(
+    name=NAME,
+    version=VERSION,
+    description=DESCRIPTION,
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    author=AUTHOR,
+    author_email=EMAIL,
+    python_requires=REQUIRES_PYTHON,
+    url=URL,
+    packages=find_packages(),
+    install_requires=[
+        "torch>=1.11.0",
+        "torchaudio",
+        "functorch",
+        "scipy",
+        "numpy",
+        "torchvision",
+        "pytorch-lightning",
+        "numba",
+        "wandb",
+    ],
+    include_package_data=True,
+    license="Apache License 2.0",
+    classifiers=[
+        "License :: OSI Approved :: Apache Software License",
+        "Topic :: Multimedia :: Sound/Audio",
+        "Topic :: Scientific/Engineering",
+    ],
+)