Update _utils.py

This commit is contained in:
Daniel Han 2025-01-16 01:18:15 -08:00
commit b99b2c5211

View file

@ -801,6 +801,7 @@ def patch_linear_scaling(
f"{model_name.title()}Attention, {model_name.title()}Config"
print(exec_code)
print(inspect.getsource(attention_module.__init__))
try:
function = inspect.getsource(attention_module.__init__)
except: