Updated "Load Model" node

2025-12-20 11:05:56 +01:00
parent 8b202a70d2
commit aba41e4d8b
1 changed files with 8 additions and 1 deletions
@@ -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,)