Spaces:

yslan
/

worldmem

Running on Zero

xizaoqu commited on Apr 10

Commit

1e18469

1 Parent(s): 982f81d

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -242,7 +242,7 @@ def generate(keys):
         memory_frames.append(new_frame)
     out_video = np.stack(memory_frames)
-    out_video = out_video.permute(0,2,3,1).numpy()
     out_video = np.clip(out_video, a_min=0.0, a_max=1.0)
     out_video = (out_video * 255).astype(np.uint8)

         memory_frames.append(new_frame)
     out_video = np.stack(memory_frames)
+    out_video = out_video.transpose(0,2,3,1)
     out_video = np.clip(out_video, a_min=0.0, a_max=1.0)
     out_video = (out_video * 255).astype(np.uint8)