xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-02 01:57:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Separate sage 1.x/2.x and sage 3.x on different functions and flags 3.

Browse Source
This commit is contained in:
Panchovix 2025-07-26 14:22:22 -04:00 committed by GitHub
parent 5e1865c1c7
commit 3c3009c020
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/model_management.py
View File

@ -1044,6 +1044,9 @@ def cast_to_device(tensor, device, dtype, copy=False):
def sage_attention_enabled():
return args.use_sage_attention
def sage_attention3_enabled():
return args.use_sage_attention3
def flash_attention_enabled():
return args.use_flash_attention