ilotoki_thu
/
alpaca_lora_4bit
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

minor fix

This commit is contained in:
John Smith 2023-04-12 13:06:30 +08:00
parent 4261bd8070
commit e946f830d4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
autograd_4bit.py
View File

@ -76,7 +76,7 @@ def switch_backend_to(to_backend):
elif to_backend == 'triton': elif to_backend == 'triton':
# detect if AutogradMatmul4bitTriton is defined # detect if AutogradMatmul4bitTriton is defined
if 'AutogradMatmul4bitTriton' not in globals(): if 'AutogradMatmul4bitTriton' not in globals():
raise ValueError('Triton not found. Please install triton_utils.') raise ValueError('Triton not found. Please install triton')
AutogradMatmul4bit = AutogradMatmul4bitTriton AutogradMatmul4bit = AutogradMatmul4bitTriton
backend = 'triton' backend = 'triton'
print(Style.BRIGHT + Fore.GREEN + 'Using Triton implementation.') print(Style.BRIGHT + Fore.GREEN + 'Using Triton implementation.')