diff --git a/nodes.py b/nodes.py
index 54bb3ab..affc416 100644
--- a/nodes.py
+++ b/nodes.py
@@ -114,7 +114,14 @@ class Trellis2LoadModel:
         
         pipeline = Trellis2ImageTo3DPipeline.from_pretrained(model_path)
         pipeline.low_vram = low_vram
-        pipeline.to(device)
+        
+        if device=="cuda":
+            if low_vram:
+                pipeline.cuda()
+            else:
+                pipeline.to(device)
+        else:
+            pipeline.to(device)
         
         return (pipeline,)