Fix: add torch to req

2024-05-29 07:06:00 +09:00
parent 9625014184
commit e66074c243
3 changed files with 5 additions and 0 deletions
--- a/requirements-colab.txt
+++ b/requirements-colab.txt
@@ -11,5 +11,7 @@ pyloudnorm
 pyopenjtalk-dict
 pypinyin
 pyworld-prebuilt
+torch
+torchaudio
 transformers
 umap-learn
--- a/requirements-infer.txt
+++ b/requirements-infer.txt
@@ -18,5 +18,6 @@ pypinyin
 pyworld-prebuilt
 # stable_ts
 # tensorboard
+torch
 transformers
 # umap-learn
--- a/requirements.txt
+++ b/requirements.txt
@@ -18,5 +18,7 @@ pypinyin
 pyworld-prebuilt
 stable_ts
 tensorboard
+torch
+torchaudio
 transformers
 umap-learn