bugfix

2026-03-18 22:08:13 +00:00 · 2025-12-20 14:00:22 +08:00
parent c6722b3f56
commit 20e1aaf908
4 changed files with 10 additions and 5 deletions
--- a/diffsynth/models/general_modules.py
+++ b/diffsynth/models/general_modules.py
@@ -87,6 +87,7 @@ class TimestepEmbeddings(torch.nn.Module):
            self.timestep_embedder = torch.nn.Sequential(
                torch.nn.Linear(dim_in, dim_out), torch.nn.SiLU(), torch.nn.Linear(dim_out, dim_out)
            )
+        self.use_additional_t_cond = use_additional_t_cond
        if use_additional_t_cond:
            self.addition_t_embedding = torch.nn.Embedding(2, dim_out)