423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-16 20:00:36 +08:00
Code Issues Packages Projects Releases Wiki Activity

tiny fix

Browse Source
This commit is contained in:
hiyouga
2024-09-01 21:15:44 +08:00
parent 64cb947c60
commit 55027282cd
2 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/llamafactory/__init__.py
View File

@@ -31,6 +31,12 @@ Dependency graph:
transformers>=4.41.2,<=4.44.3
packing:
transformers>=4.41.2,<=4.44.3
Disable version checking: DISABLE_VERSION_CHECK=1
Enable VRAM recording: RECORD_VRAM=1
Force check imports: FORCE_CHECK_IMPORTS=1
Force using torchrun: FORCE_TORCHRUN=1
Use modelscope: USE_MODELSCOPE_HUB=1
"""
from .extras.env import VERSION