use latest triton

yousufmo · Feb 16, 2023 · b67f8d8 · b67f8d8
1 parent be281c1
commit b67f8d8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -51,7 +51,7 @@ opt.step()
 opt.zero_grad()
 ```
 
-To use a fused kernel for updating the parameters, first `pip install triton`, then
+To use a fused kernel for updating the parameters, first `pip install triton -U --pre`, then
 
 ```python
 opt = Lion(

diff --git a/lion_pytorch/triton.py b/lion_pytorch/triton.py
@@ -4,7 +4,7 @@
     import triton
     import triton.language as tl
 except ImportError as e:
-    print('triton is not installed, please install by running `pip install triton`')
+    print('triton is not installed, please install by running `pip install triton -U --pre`')
     exit()
 
 @triton.jit