From e66074c243496d538e403686ab71b4944ec28e85 Mon Sep 17 00:00:00 2001 From: litagin02 Date: Wed, 29 May 2024 07:06:00 +0900 Subject: [PATCH] Fix: add torch to req --- requirements-colab.txt | 2 ++ requirements-infer.txt | 1 + requirements.txt | 2 ++ 3 files changed, 5 insertions(+) diff --git a/requirements-colab.txt b/requirements-colab.txt index 2f92ef1..93c086c 100644 --- a/requirements-colab.txt +++ b/requirements-colab.txt @@ -11,5 +11,7 @@ pyloudnorm pyopenjtalk-dict pypinyin pyworld-prebuilt +torch +torchaudio transformers umap-learn diff --git a/requirements-infer.txt b/requirements-infer.txt index dcae43f..e70e757 100644 --- a/requirements-infer.txt +++ b/requirements-infer.txt @@ -18,5 +18,6 @@ pypinyin pyworld-prebuilt # stable_ts # tensorboard +torch transformers # umap-learn diff --git a/requirements.txt b/requirements.txt index 6695153..3ae7567 100644 --- a/requirements.txt +++ b/requirements.txt @@ -18,5 +18,7 @@ pypinyin pyworld-prebuilt stable_ts tensorboard +torch +torchaudio transformers umap-learn