cuda trellis

app.py

@@ -67,7 +67,7 @@ def preprocess_image(image: Image.Image) -> Image.Image:
     if trellis is None:
         # If the pipeline is not loaded, just return the original image
         return image
-    trellis.cuda()
+    # trellis.cuda()
     processed_image = trellis.preprocess_image(image)
     return processed_image
 
@@ -125,13 +125,6 @@ def get_trellis_pipeline():
         try:
             print("Loading Trellis pipeline...")
             _trellis_pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
-            # _trellis_pipeline.cuda()
-            
-            # # Preload rembg by processing a small test image
-            # try:
-            #     _trellis_pipeline.preprocess_image(Image.fromarray(np.zeros((512, 512, 3), dtype=np.uint8)))
-            # except Exception as e:
-            #     print(f"Warning when preloading rembg: {e}")
         except Exception as e:
             print(f"Error loading Trellis pipeline: {e}")
             return None
@@ -331,6 +324,9 @@ def image_to_3d(
         if pipeline is None:
             return None, "Trellis pipeline is unavailable."
         
+        # Call cuda() here in the GPU worker process
+        pipeline.cuda()
+        
         outputs = pipeline.run(
             image,
             seed=seed,