nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix sp i2v

Browse Source
This commit is contained in:
SolitaryThinker
2025-06-29 19:46:08 -07:00
parent 9ab7f031af
commit 7dcfe4ea8f
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastvideo/v1/training/training_pipeline.py
+5 -4
View File
@@ -233,6 +233,11 @@ class TrainingPipeline(ComposedPipelineBase, ABC):
training_batch.sigmas = sigmas
training_batch.noise = noise
# shard noisy_model_input to match
training_batch.noisy_model_input = shard_latents_across_sp(
training_batch.noisy_model_input,
num_latent_t=self.training_args.num_latent_t)
return training_batch
def _build_attention_metadata(
@@ -367,10 +372,6 @@ class TrainingPipeline(ComposedPipelineBase, ABC):
training_batch.latents = shard_latents_across_sp(
training_batch.latents,
num_latent_t=self.training_args.num_latent_t)
# shard noisy_model_input to match
training_batch.noisy_model_input = shard_latents_across_sp(
training_batch.noisy_model_input,
num_latent_t=self.training_args.num_latent_t)
# shard noise to match latents
training_batch.noise = shard_latents_across_sp(
training_batch.noise,