Spaces:

tencent
/

DepthCrafter

Running on Zero

wbhu-tc commited on Sep 20, 2024

Commit

173e979

1 Parent(s): a2875f5

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,6 @@ pipe = DepthCrafterPipeline.from_pretrained(
     variant="fp16",
 )
 pipe.to("cuda")
-pipe.enable_xformers_memory_efficient_attention()
 @spaces.GPU(duration=120)
@@ -53,6 +52,7 @@ def infer_depth(
     save_npz: bool = False,
 ):
     set_seed(seed)
     frames, target_fps = read_video_frames(video, process_length, target_fps, max_res)
     print(f"==> video name: {video}, frames shape: {frames.shape}")

     variant="fp16",
 )
 pipe.to("cuda")
 @spaces.GPU(duration=120)
     save_npz: bool = False,
 ):
     set_seed(seed)
+    pipe.enable_xformers_memory_efficient_attention()
     frames, target_fps = read_video_frames(video, process_length, target_fps, max_res)
     print(f"==> video name: {video}, frames shape: {frames.shape}")