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