CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

triton

Browse Source
This commit is contained in:
Haoming 2026-06-01 12:05:25 +08:00
parent b78c007347
commit 5d14e0c901
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/args.py
View File

@ -103,6 +103,7 @@ parser.add_argument("--mmap-torch-files", action="store_true", help="Use mmap wh
parser.add_argument("--disable-mmap", action="store_true", help="Don't use mmap when loading safetensors")
parser.add_argument("--tiled-conv2d", type=int, default=0, metavar="TILE_SIZE", choices=[0, 64, 128, 256, 512], help="reduce VAE memory usage ; increase processing time")
parser.add_argument("--enable-triton-backend", action="store_true", help="Enable the use of Triton backend in comfy-kitchen")
class SageAttentionFuncs(enum.Enum):

5
backend/quant_ops.py
View File

@ -22,9 +22,10 @@ else:
if cuda_version < (13,):
ck.registry.disable("cuda")
from backend.args import args
# https://github.com/Comfy-Org/ComfyUI/blob/v0.16.4/comfy/quant_ops.py#L24
ck.registry.disable("triton")
if not args.enable_triton_backend:
ck.registry.disable("triton")
# region FP8 Layouts