multimodalart HF Staff commited on
Commit
3947f33
·
verified ·
1 Parent(s): 2a3fbc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -365,7 +365,7 @@ def generate(prompt,
365
  # We expect B=1 from typical generation
366
  video_single_batch = result_frames_tensor[0] # Shape: (C, F, H, W)
367
  video_single_batch = (video_single_batch / 2 + 0.5).clamp(0, 1) # Normalize to [0,1]
368
- video_single_batch = video_single_batch.permute(1, 2, 3, 0).cpu().numpy() # F, H, W, C
369
 
370
  for frame_idx in range(video_single_batch.shape[0]):
371
  frame_np = (video_single_batch[frame_idx] * 255).astype(np.uint8)
 
365
  # We expect B=1 from typical generation
366
  video_single_batch = result_frames_tensor[0] # Shape: (C, F, H, W)
367
  video_single_batch = (video_single_batch / 2 + 0.5).clamp(0, 1) # Normalize to [0,1]
368
+ video_single_batch = video_single_batch.permute(1, 2, 3, 0).cpu().float().numpy() # F, H, W, C
369
 
370
  for frame_idx in range(video_single_batch.shape[0]):
371
  frame_np = (video_single_batch[frame_idx] * 255).astype(np.uint8)