Ruff fixes

This commit is contained in:
Panchovix 2025-07-25 21:18:42 -04:00 committed by GitHub
parent 2d69b2f064
commit bb7393565b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -36,10 +36,7 @@ if model_management.sage_attention_enabled():
pass
if not sage_attention_available:
if e.name == "sageattention":
logging.error(f"\n\nTo use the `--use-sage-attention` feature, the `sageattention` package must be installed first.\ncommand:\n\t{sys.executable} -m pip install sageattention")
else:
raise e
logging.error(f"\n\nTo use the `--use-sage-attention` feature, the `sageattention` package must be installed first.\ncommand:\n\t{sys.executable} -m pip install sageattention")
exit(-1)
if model_management.flash_attention_enabled():