support galore

This commit is contained in:
hiyouga committed 2024-03-07 22:41:36 +08:00
1 parent 725f7cd70f
commit 28f7862188
12 files changed
+115 -16

No files matched your search

+4
View File
@@ -21,6 +21,10 @@ def is_flash_attn2_available():
return _is_package_available("flash_attn") and _get_package_version("flash_attn").startswith("2")
def is_galore_available():
return _is_package_available("galore_torch")
def is_jieba_available():
return _is_package_available("jieba")