From 3359aa0d48cdf82b9709e075019519003571799f Mon Sep 17 00:00:00 2001
From: ExponentialML <59846140+ExponentialML@users.noreply.github.com>
Date: Thu, 21 Mar 2024 15:18:24 -0700
Subject: [PATCH] Disable AD

This will be done in the future as it needs more testing.
---
 utils/attention_functions.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/utils/attention_functions.py b/utils/attention_functions.py
index f1fad7f..c2b2a45 100644
--- a/utils/attention_functions.py
+++ b/utils/attention_functions.py
@@ -41,7 +41,7 @@ class VisualStyleProcessor(object):
         module_self, 
         keys_scale: float = 1.0,
         enabled: bool = True, 
-        enabled_animatediff: bool = True,
+        enabled_animatediff: bool = False,
         adain_queries: bool = True,
         adain_keys: bool = True,
         adain_values: bool = False 
@@ -81,6 +81,7 @@ class VisualStyleProcessor(object):
             out = optimized_attention_masked(q, k, v, self.module_self.heads, mask)
         return self.module_self.to_out(out)
 
+    # TODO
     def visual_style_forward_ad(self, x, context=None, value=None, mask=None):
         q = module_self.to_q(x)
         context = default(context, x)