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

update data processors

Browse Source
This commit is contained in:
hiyouga
2024-06-07 04:15:40 +08:00
parent 181dbb0d05
commit ccc8b64cc2
6 changed files with 190 additions and 139 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/cli.py
View File

@@ -72,7 +72,7 @@ def main():
elif command == Command.EXPORT:
export_model()
elif command == Command.TRAIN:
disable_torchrun = os.environ.get("DISABLE_TORCHRUN", "0").lower() in ["true", "1"]
disable_torchrun = os.environ.get("TORCHRUN_DISABLED", "0").lower() in ["true", "1"]
if disable_torchrun and get_device_count() > 1:
logger.warning("`torchrun` cannot be disabled when device count > 1.")
disable_torchrun = False