Spaces:

pengHTYX
/

Era3D_MV_demo

Runtime error

pengHTYX commited on May 29, 2024

Commit

67c0098

•

1 Parent(s): d882e8b

'update'

Files changed (1) hide show

app.py CHANGED Viewed

@@ -298,7 +298,7 @@ def run_demo():
     torch.set_grad_enabled(False)
     print(1)
     pipeline.to(device=f'cuda:{_GPU_ID}')
-    pipeline.unet.enable_xformers_memory_efficient_attention()
     print(2)
     predictor = sam_init()

     torch.set_grad_enabled(False)
     print(1)
     pipeline.to(device=f'cuda:{_GPU_ID}')
+    # pipeline.unet.enable_xformers_memory_efficient_attention()
     print(2)
     predictor = sam_init()