fix version typo

mosaicml · Oct 10, 2023 · faa562d · faa562d
1 parent 64e2f03
commit faa562d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -95,7 +95,7 @@
     'flash-attn==2.3.2',
     'mosaicml-turbo==0.0.4',
     # PyPI does not support direct dependencies, so we remove this line before uploading from PyPI
-    'xentropy-cuda-lib@git+https://github.com/HazyResearch/[email protected].1#subdirectory=csrc/xentropy',
+    'xentropy-cuda-lib@git+https://github.com/HazyResearch/[email protected].2#subdirectory=csrc/xentropy',
 ]
 
 extra_deps['peft'] = [