From e946f830d4c0f5aa3fef89428d3f5e8a46aeb023 Mon Sep 17 00:00:00 2001 From: John Smith Date: Wed, 12 Apr 2023 13:06:30 +0800 Subject: [PATCH] minor fix --- autograd_4bit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/autograd_4bit.py b/autograd_4bit.py index 544e429..476da12 100644 --- a/autograd_4bit.py +++ b/autograd_4bit.py @@ -76,7 +76,7 @@ def switch_backend_to(to_backend): elif to_backend == 'triton': # detect if AutogradMatmul4bitTriton is defined if 'AutogradMatmul4bitTriton' not in globals(): - raise ValueError('Triton not found. Please install triton_utils.') + raise ValueError('Triton not found. Please install triton') AutogradMatmul4bit = AutogradMatmul4bitTriton backend = 'triton' print(Style.BRIGHT + Fore.GREEN + 'Using Triton implementation.')