update comment

This commit is contained in:
Aryan
2024-12-23 14:01:39 +01:00
parent 75a854f9df
commit 78712f60a5
2 changed files with 5 additions and 2 deletions
+4 -1
View File
@@ -151,7 +151,10 @@ def prepare_latents(
else: else:
h = vae._encode(image_or_video) h = vae._encode(image_or_video)
_, _, num_frames, height, width = h.shape _, _, num_frames, height, width = h.shape
# TODO(aryan): this is very very very stupid, but anything to make it work for now. refactor and design better later
# TODO(aryan): This is very stupid that we might possibly be storing the latents_mean and latents_std in every file
# if precomputation is enabled. We should probably have a single file where re-usable properties like this are stored
# so as to reduce the disk memory requirements of the precomputed files.
return { return {
"latents": h, "latents": h,
"num_frames": num_frames, "num_frames": num_frames,