Update attn_processor.py

2025-03-26 08:35:14 +08:00
parent 57b3421de3
commit 908345809b
1 changed files with 3 additions and 2 deletions
@@ -68,7 +68,8 @@ class PersonalizeAnythingAttnProcessor:
        ###################################################################################
        if timestep is not None:
            timestep = timestep
-            
+        else: #timestep is None
+            timestep = 0.0
        concept_process = self.concept_process # token concatenation
        c_q = concept_process and True # if token concatenation is applied to q
        c_kv = concept_process and True  # if token concatenation is applied to kv
@@ -442,4 +443,4 @@ class MultiPersonalizeAnythingAttnProcessor:
            hidden_states = hidden_states[:, :dims, :]
            ################################################################
            
-            return hidden_states
+            return hidden_states