Merge pull request #24 from stealthinu/onnx化

PyInstaller化とONNX対応化の修正
isletennos · Dec 13, 2022 · 817da67 · 817da67
2 parents 46f2a4c + aa45110
commit 817da67
Show file tree

Hide file tree

Showing 29 changed files with 812 additions and 24,853 deletions.
diff --git a/python/FixBuildPlugin_pytorch.py b/python/FixBuildPlugin_pytorch.py
diff --git a/python/Pipfile b/python/Pipfile
@@ -0,0 +1,23 @@
+[[source]]
+url = "https://pypi.org/simple"
+verify_ssl = true
+name = "pypi"
+
+[[source]]
+url = "https://download.pytorch.org/whl/cu111/"
+verify_ssl = true
+name = "downloadpytorch"
+
+[packages]
+torch = {version = "==1.8.1+cu111", index = "downloadpytorch"}
+noisereduce = "==2.0.0"
+numpy = "==1.21.1"
+scikit-learn = "==1.0.2"
+sounddevice = "==0.4.4"
+SoundFile = "==0.10.3.post1"
+onnxruntime-directml = "1.13.1"
+
+[dev-packages]
+
+[requires]
+python_version = "3.9"
diff --git a/python/attentions.py b/python/attentions.py