parent 03845a5

author Will <[email protected]> 1728870664 -0400 committer Will <[email protected]> 1732659637 -0500 parent 03845a5 author Will <[email protected]> 1728870664 -0400 committer Will <[email protected]> 1732659617 -0500 parent 03845a5 author Will <[email protected]> 1728870664 -0400 committer Will <[email protected]> 1732659576 -0500 parent 03845a5 author Will <[email protected]> 1728870664 -0400 committer Will <[email protected]> 1732659547 -0500 parent 03845a5 author Will <[email protected]> 1728870664 -0400 committer Will <[email protected]> 1732659503 -0500 parent 03845a5 author Will <[email protected]> 1728870664 -0400 committer Will <[email protected]> 1732659433 -0500 Major Refactor
will-rice · Nov 26, 2024 · 931d70e · 931d70e
1 parent 03845a5
commit 931d70e
Show file tree

Hide file tree

Showing 57 changed files with 4,345 additions and 1,681 deletions.
diff --git a/.gitignore b/.gitignore
@@ -134,3 +134,6 @@ logs/
 
 # ruff
 .ruff_cache/
+
+# mac
+.DS_Store
diff --git a/.mypy.ini b/.mypy.ini
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -12,29 +12,18 @@ repos:
     hooks:
       - id: prettier
         files: \.(css|less|scss|json|yaml|yml|html|md|mdx|js|jsx|ts|tsx|graphql|gql|vue|svelte|toml)$
-  - repo: https://github.com/astral-sh/ruff-pre-commit
-    # Ruff version.
-    rev: v0.1.3
-    hooks:
-      - id: ruff
-        args: ["--fix"]
   - repo: local
     hooks:
-      - id: black
-        name: black
-        language: system
-        entry: black
-        types: [python]
-      - id: isort
-        name: isort
+      - id: ruff
+        name: ruff
         language: system
-        entry: isort
-        args: ["--profile", "black"]
-        types: [python]
+        entry: ruff check . --fix
+        pass_filenames: false
+        always_run: true
       - id: mypy
         name: mypy
         language: system
-        entry: mypy denoisers
+        entry: mypy .
         pass_filenames: false
         always_run: true
       - id: pytest

diff --git a/README.md b/README.md
@@ -6,13 +6,14 @@ Denoisers is a denoising library for audio with a focus on simplicity and ease o
 
 [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/wrice/denoisers)
 
-
 ## Usage/Examples
 
 ```sh
 pip install denoisers
 ```
 
+### Inference
+
 ```python
 import torch
 import torchaudio
@@ -43,3 +44,19 @@ for i in tqdm(range(0, padded.shape[-1], chunk_size)):
 
 denoised = torch.concat(clean, 1)[:, :audio.shape[-1]]
 ```
+
+### Train
+
+```sh
+
+train unet1d unet1d-24khz /data_root/
+
+```
+
+### Publish
+
+```sh
+
+publish model model_name /path/to/model
+
+```
diff --git a/bin/run b/bin/run
@@ -2,7 +2,6 @@
 
 set -e
 
-subsys=$1
-command=$2
-shift; shift
-exec python -m "denoisers.scripts.$subsys.$command" "$@"
+command=$1
+shift
+exec python -m "denoisers.scripts.$command" "$@"
diff --git a/denoisers/datamodules/__init__.py b/denoisers/datamodules/__init__.py
diff --git a/denoisers/datamodules/unet.py b/denoisers/datamodules/unet.py
diff --git a/denoisers/datamodules/unet1d.py b/denoisers/datamodules/unet1d.py
-Original file line number
+Diff line change
@@ Expand Up / @@ -134,3 +134,6 @@ logs/ @@
     # ruff
     .ruff_cache/
+    # mac
+    .DS_Store