Skip to content

Commit 34523f8

Browse files
committed
[Fix] dataloader issue, distill for single gpu ready
1 parent 606b642 commit 34523f8

File tree

4 files changed

+509
-517
lines changed

4 files changed

+509
-517
lines changed

fastvideo/v1/dataset/parquet_datasets.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@ def __init__(self,
119119
plan = json.load(f)
120120
self.neg_metadata = plan["negative_prompt"][0]
121121

122+
# Add unconditional embeddings for distillation (like in LatentDataset)
122123
self.uncond_prompt_embed = torch.zeros(512, 4096).to(torch.float32)
123124
self.uncond_prompt_mask = torch.zeros(1, 512).bool()
124125

0 commit comments

Comments
 (0)