diff --git a/text/__init__.py b/text/__init__.py
index 8dd10db..8b3ed9d 100644
--- a/text/__init__.py
+++ b/text/__init__.py
@@ -1,5 +1,5 @@
 from text.symbols import *
-
+import torch
 
 _symbol_to_id = {s: i for i, s in enumerate(symbols)}