Update src_inference/lora_helper.py

src_inference/lora_helper.py

@@ -185,6 +185,10 @@ def unset_lora(transformer):
         lora_attn_procs[name] = FluxAttnProcessor2_0()
     transformer.set_attn_processor(lora_attn_procs)
 
+def clear_cache(transformer):
+    for name, attn_processor in transformer.attn_processors.items():
+        if hasattr(attn_processor, "bank_kv"):
+            attn_processor.bank_kv.clear()
 
 '''
 unset_lora(pipe.transformer)