storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-22 08:05:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

update comment

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/ltx_video/ltx_video_lora.py
+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,
finetrainers/trainer.py
+1 -1
View File
@@ -813,7 +813,7 @@ class Trainer:
num_videos_per_prompt=self.args.num_validation_videos_per_prompt, num_videos_per_prompt=self.args.num_validation_videos_per_prompt,
generator=self.state.generator, generator=self.state.generator,
) )
# Remove all hooks that might have been added during pipeline initialization to the models # Remove all hooks that might have been added during pipeline initialization to the models
pipeline.remove_all_hooks() pipeline.remove_all_hooks()
del pipeline del pipeline