Downloads · 30 days
0
jmig1109/comfyui-attn-wheels-cuda13
comfyui-attn-wheels-cuda13 is a machine learning model from jmig1109. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as unknown.
Flash Attention and Sage Attention Wheels to use in combination with YanWenKun/ComfyUI-Docker ComfyUI images. The wheels have been built to work with the cu130-slim image in particular, using Cuda 13, Python 3.13, and…
Downloads · 30 days
0
Access
Public
Updated Feb 27, 2026
Repo size
265 MB
Likes
2
Public
Click a slice to open those files.
.whl265 MB · 100%
From the Hugging Face model README
Flash Attention and Sage Attention Wheels to use in combination with YanWenKun/ComfyUI-Docker ComfyUI images. The wheels have been built to work with the cu130-slim image in particular, using Cuda 13, Python 3.13, and Pytorch 2.10 on Linux. My system uses a mix of Ampere and Blackwell GPUs.
To use the wheels, place them in the root folder of your container, and update the pre-start script according to your needs. In your docker compose, use the cli-args to either --use-flash-attention or --use-sage-attention
WARNING: The Sage Attention 3 wheel has been built specifically for blackwell gpus. It also uses a different package name than previous sage attention version and cannot be activated by CLI Arguments, only using custom nodes. Many models are still unsupported.