xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-06-09 17:44:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-11-22 17:12:06 +02:00
parent fb5aa296ae
commit 44620cb566
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes/nodes.py
View File

@ -2172,10 +2172,11 @@ class CheckpointLoaderKJ:
def patch(self, ckpt_name, patch_cublaslinear, sage_attention): def patch(self, ckpt_name, patch_cublaslinear, sage_attention):
from comfy.ops import disable_weight_init, CastWeightBiasOp, cast_bias_weight from comfy.ops import disable_weight_init, CastWeightBiasOp, cast_bias_weight
from nodes import CheckpointLoaderSimple from nodes import CheckpointLoaderSimple
try: if patch_cublaslinear:
from cublas_ops import CublasLinear try:
except ImportError: from cublas_ops import CublasLinear
raise Exception("Can't import 'torch-cublas-hgemm', install it from here https://github.com/aredden/torch-cublas-hgemm") except ImportError:
raise Exception("Can't import 'torch-cublas-hgemm', install it from here https://github.com/aredden/torch-cublas-hgemm")
if sage_attention: if sage_attention:
from sageattention import sageattn from sageattention import sageattn