From 196401e770b62f780cec1e1df53acce3f54f8225 Mon Sep 17 00:00:00 2001
From: patientx <patient.x@gmail.com>
Date: Thu, 15 May 2025 14:03:07 +0300
Subject: [PATCH] Update install-n.bat

---
 install-n.bat | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/install-n.bat b/install-n.bat
index e7240382b..878518090 100644
--- a/install-n.bat
+++ b/install-n.bat
@@ -64,6 +64,11 @@ pip install flash_attn-2.7.4.post1-py3-none-any.whl --quiet
 del fa.zip
 del flash_attn-2.7.4.post1-py3-none-any.whl
 
+echo  ::  %time:~0,8%  ::  - Patching sage-attention
+copy comfy\customzluda\sa\quant_per_block.py venv\Lib\site-packages\sageattention\quant_per_block.py /y >NUL
+copy comfy\customzluda\sa\attn_qk_int8_per_block_causal.py venv\Lib\site-packages\sageattention\attn_qk_int8_per_block_causal.py /y >NUL
+copy comfy\customzluda\sa\attn_qk_int8_per_block.py venv\Lib\site-packages\sageattention\attn_qk_int8_per_block.py /y >NUL
+
 echo.
 echo  ::  %time:~0,8%  ::  Custom node(s) installation ...
 echo.