diff --git a/src/attn_processor.py b/src/attn_processor.py index ff4ddd6..c950089 100644 --- a/src/attn_processor.py +++ b/src/attn_processor.py @@ -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 \ No newline at end of file + return hidden_states