From 44620cb566bae67dcbcdbf5b942c644c102193f8 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 22 Nov 2024 17:12:06 +0200 Subject: [PATCH] Update nodes.py --- nodes/nodes.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/nodes/nodes.py b/nodes/nodes.py index 44dc12e..7422db3 100644 --- a/nodes/nodes.py +++ b/nodes/nodes.py @@ -2172,10 +2172,11 @@ class CheckpointLoaderKJ: def patch(self, ckpt_name, patch_cublaslinear, sage_attention): from comfy.ops import disable_weight_init, CastWeightBiasOp, cast_bias_weight from nodes import CheckpointLoaderSimple - try: - from cublas_ops import CublasLinear - except ImportError: - raise Exception("Can't import 'torch-cublas-hgemm', install it from here https://github.com/aredden/torch-cublas-hgemm") + if patch_cublaslinear: + try: + from cublas_ops import CublasLinear + except ImportError: + raise Exception("Can't import 'torch-cublas-hgemm', install it from here https://github.com/aredden/torch-cublas-hgemm") if sage_attention: from sageattention import sageattn