diff --git a/nodes.py b/nodes.py
index f4a3ced..e34111c 100644
--- a/nodes.py
+++ b/nodes.py
@@ -12,6 +12,7 @@ import comfy.model_patcher
 import comfy.model_management
 import folder_paths
 import nodes
+import torch
 
 from .ops import GGMLOps, move_patch_to_device
 from .loader import gguf_sd_loader, gguf_clip_loader
diff --git a/pyproject.toml b/pyproject.toml
index f1498fe..6326744 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "smartmodelloaders-mxd"
 description = "Smart, unified model loaders for ComfyUI that support both standard .safetensors and quantized .gguf formats — no switching nodes required. Includes flexible UNET and CLIP loaders that work across models like SDXL, SD3, Flux, and more."
-version = "1.0.3"
+version = "1.0.4"
 license = {file = "LICENSE"} 
 # classifiers = [
 #     # For OS-independent nodes (works on all operating systems)