cunliangkong/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-03-22 01:43:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

[misc] lint code (#9395)

Browse Source
This commit is contained in:
Yaowei Zheng
2025-11-03 22:08:59 +08:00
committed by GitHub
parent 215580c77d
commit 3ae15da9c0
17 changed files with 82 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/llamafactory/launcher.py
View File

@@ -54,8 +54,7 @@ def launch():
)
command = sys.argv.pop(1) if len(sys.argv) > 1 else "help"
if is_env_enabled("USE_MCA"):
# force use torchrun
if is_env_enabled("USE_MCA"): # force use torchrun
os.environ["FORCE_TORCHRUN"] = "1"
if command == "train" and (is_env_enabled("FORCE_TORCHRUN") or (get_device_count() > 1 and not use_ray())):