transformers - 💡(How to fix) Fix AttentionInterface.register changes behavior of registered function

from transformers import AutoModelForCausalLM, AttentionInterface, pipeline
from transformers.integrations.sdpa_attention import sdpa_attention_forward

model1 = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B", attn_implementation="sdpa")
pipeline1 = pipeline(task="text-generation", model=model1, tokenizer="meta-llama/Llama-3.2-1B",
                    cache_implementation="static")
print(pipeline1("It was a bright cold day in April, and the clocks were striking thirteen."))

AttentionInterface.register("reregistered_sdpa", sdpa_attention_forward)
model2 = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B", attn_implementation="reregistered_sdpa")
pipeline2 = pipeline(task="text-generation", model=model2, tokenizer="meta-llama/Llama-3.2-1B",
                     cache_implementation="static")
print(pipeline2("It was a bright cold day in April, and the clocks were striking thirteen."))