mirror of
https://github.com/hiyouga/LlamaFactory.git
synced 2026-03-23 10:43:22 +08:00
[data] fix mm pluigin for qwen omni video training (#9388)
Co-authored-by: frozenleaves <frozen@Mac.local>
This commit is contained in:
@@ -57,7 +57,7 @@ def launch():
|
||||
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())):
|
||||
# launch distributed training
|
||||
nnodes = os.getenv("NNODES", "1")
|
||||
|
||||
Reference in New Issue
Block a user