diff --git a/nodes_model_loading.py b/nodes_model_loading.py index 4711a5a9..aac1062a 100644 --- a/nodes_model_loading.py +++ b/nodes_model_loading.py @@ -1124,9 +1124,11 @@ def loadmodel(self, model, base_precision, load_device, quantization, if "sage" in attention_mode: try: from sageattention import sageattn - except Exception as e: - raise ValueError(f"Can't import SageAttention: {str(e)}") - + use_sageattn = True + except ImportError as e: + print(f"SageAttention is not available ->falling back to default attention: {e}") + use_sageattn = False + gguf = False if model.endswith(".gguf"): if quantization != "disabled":