move trellis cuda

app.py

@@ -67,6 +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()
     processed_image = trellis.preprocess_image(image)
     return processed_image
 
@@ -124,13 +125,13 @@ def get_trellis_pipeline():
         try:
             print("Loading Trellis pipeline...")
             _trellis_pipeline = TrellisImageTo3DPipeline.from_pretrained("JeffreyXiang/TRELLIS-image-large")
-            _trellis_pipeline.cuda()
+            # _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}")
+            # # 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