1

app.py

@@ -54,7 +54,7 @@ os.makedirs(args.cache_dir, exist_ok=True)
 
 geometry_model = Step1X3DGeometryPipeline.from_pretrained(
     "stepfun-ai/Step1X-3D", subfolder=args.geometry_model
-)
+).to("cuda")
 
 texture_model = Step1X3DTexturePipeline.from_pretrained("stepfun-ai/Step1X-3D", subfolder=args.texture_model)
 
@@ -63,7 +63,7 @@ texture_model = Step1X3DTexturePipeline.from_pretrained("stepfun-ai/Step1X-3D",
 def generate_func(
     input_image_path, guidance_scale, inference_steps, max_facenum, symmetry, edge_type
 ):
-    geometry_model = geometry_model.to("cuda")
+    # geometry_model = geometry_model.to("cuda")
     if "Label" in args.geometry_model:
         symmetry_values = ['x', 'asymmetry']
         out = geometry_model(