change print to logging.info

This commit is contained in:
Panchovix 2025-07-25 21:24:47 -04:00 committed by GitHub
parent 9a70423973
commit 27720a1e45
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -25,13 +25,13 @@ if model_management.sage_attention_enabled():
from sageattn import sageattn_blackwell
SAGE_ATTENTION_3_AVAILABLE = True
sage_attention_available = True
print("Found SageAttention3 (sageattn package)")
logging.info("Found SageAttention3 (sageattn package)")
except ImportError:
try:
from sageattention import sageattn
sage_attention_available = True
print("Found SageAttention2 (sageattention package)")
logging.info("Found SageAttention2 (sageattention package)")
except ModuleNotFoundError as e:
pass