ModelMan

Runtime error

wyysf commited on May 31, 2024

Commit

ef66f74

verified ·

1 Parent(s): 9bc9474

Update gradio_app.py

Files changed (1) hide show

gradio_app.py CHANGED Viewed

@@ -145,7 +145,7 @@ def gen_mvimg(
     elif mvimg_model == "Era3D":
         global era3d_pipeline
-        era3d_pipeline.to(device=device)
         era3d_pipeline.unet.enable_xformers_memory_efficient_attention()
         crop_size = 420

     elif mvimg_model == "Era3D":
         global era3d_pipeline
+        era3d_pipeline.to(device)
         era3d_pipeline.unet.enable_xformers_memory_efficient_attention()
         crop_size = 420