Hugging Face's logo

Spaces:
tencent
/
DepthCrafter
Running on Zero

App Files Community
wbhu-tc commited on
Commit
a2875f5
·
1 Parent(s): 6654f6a

update

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -33,6 +33,7 @@ pipe = DepthCrafterPipeline.from_pretrained(
33
  variant="fp16",
34
  )
35
  pipe.to("cuda")
 
36
 
37
 
38
  @spaces.GPU(duration=120)
 
33
  variant="fp16",
34
  )
35
  pipe.to("cuda")
36
+ pipe.enable_xformers_memory_efficient_attention()
37
 
38
 
39
  @spaces.GPU(duration=120)