423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2026-03-08 04:35:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

support galore

Browse Source 
Former-commit-id: 28f7862188
This commit is contained in:
hiyouga
2024-03-07 22:41:36 +08:00
parent 1af71f548c
commit 2c010c72b8
12 changed files with 115 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/extras/packages.py
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")