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

[3rdparty] fix redundant process group destroy for ray (#7395)

Browse Source 
* fix redundant process group destroy for ray

* Update tuner.py

---------

Co-authored-by: hoshi-hiyouga <hiyouga@buaa.edu.cn>
This commit is contained in:
Eric Tang 2025-03-20 19:56:47 -07:00 committed by GitHub
parent 555b71a1cb
commit d8a5571be7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llamafactory/train/tuner.py
View File

@ -38,6 +38,7 @@ from .trainer_utils import get_ray_trainer, get_swanlab_callback
if is_ray_available(): if is_ray_available():
import ray
from ray.train.huggingface.transformers import RayTrainReportCallback from ray.train.huggingface.transformers import RayTrainReportCallback
@ -77,6 +78,9 @@ def _training_function(config: dict[str, Any]) -> None:
else: else:
raise ValueError(f"Unknown task: {finetuning_args.stage}.") raise ValueError(f"Unknown task: {finetuning_args.stage}.")
if is_ray_available() and ray.is_initialized():
return # if ray is intialized it will destroy the process group on return
try: try:
if dist.is_initialized(): if dist.is_initialized():
dist.destroy_process_group() dist.destroy_process_group()