From a1fa27b069b4ccb4efa07fd0979485d136562e41 Mon Sep 17 00:00:00 2001
From: ExponentialML <59846140+ExponentialML@users.noreply.github.com>
Date: Thu, 21 Mar 2024 20:59:42 -0700
Subject: [PATCH] Update scale_latents

---
 nodes.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/nodes.py b/nodes.py
index bcaaff6..f340297 100644
--- a/nodes.py
+++ b/nodes.py
@@ -205,8 +205,13 @@ class DynamiCrafterProcessor:
 
         c_concat = encoded_latent
         c_concat = model.model.process_latent_in(c_concat)
+        
         if scale_latents:
-            c_concat *= 1.32
+            vae_process_input = vae.process_input
+            vae.process_input = lambda image: (image - .5) * 2
+            c_concat = vae.encode(images[:, :, :, :3])
+            vae.process_input = vae_process_input
+            c_concat = model.model.process_latent_in(c_concat)
             
         fs = torch.tensor([fps], dtype=torch.long, device=model_management.intermediate_device())