Fabrice-TIERCELIN commited on
Commit
9382ddb
·
verified ·
1 Parent(s): 6dd63f2
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -994,7 +994,7 @@ def worker_video(input_video, end_frame, prompts, n_prompt, seed, batch, resolut
994
  image_encoder=image_encoder, feature_extractor=feature_extractor, device=gpu
995
  )[:2]
996
  end_latent = end_latent.to(dtype=torch.float32, device=cpu)
997
- end_latent = end_latent.repeat(-1, -1, 100, -1, -1)
998
  else:
999
  end_latent = end_clip_embedding = None
1000
 
 
994
  image_encoder=image_encoder, feature_extractor=feature_extractor, device=gpu
995
  )[:2]
996
  end_latent = end_latent.to(dtype=torch.float32, device=cpu)
997
+ end_latent = end_latent.expand(-1, -1, 100, -1, -1)
998
  else:
999
  end_latent = end_clip_embedding = None
1000