diff --git a/nodes_literals.py b/nodes_literals.py
index fcb5fe1..8692599 100644
--- a/nodes_literals.py
+++ b/nodes_literals.py
@@ -1,6 +1,7 @@
 from sys import float_info
 from typing import Any
 from nodes import MAX_RESOLUTION
+import torch
 
 class SeedGenerator:
     RETURN_TYPES = ("INT",)