Added training for Wan2.1. Not finalized, wait.

2026-04-29 18:51:37 +00:00 · 2025-03-07 13:53:44 -07:00
parent 4e3bda7c70
commit 391cf80fea
7 changed files with 393 additions and 50 deletions
--- a/toolkit/models/base_model.py
+++ b/toolkit/models/base_model.py
@@ -625,6 +625,15 @@ class BaseModel:
        )
        noise = apply_noise_offset(noise, noise_offset)
        return noise
+    
+    def get_latent_noise_from_latents(
+        self,
+        latents: torch.Tensor,
+        noise_offset=0.0
+    ):
+        noise = torch.randn_like(latents)
+        noise = apply_noise_offset(noise, noise_offset)
+        return noise

    def add_noise(
            self,