chore: import upstream snapshot with attribution
Lint test / lint (push) Has been cancelled

This commit is contained in:
wehub-resource-sync
2026-07-13 13:34:58 +08:00
commit a203934033
1368 changed files with 175001 additions and 0 deletions
+18
View File
@@ -0,0 +1,18 @@
# Copyright (c) ModelScope Contributors. All rights reserved.
from .arguments import RayArguments
from .base import RayHelper
def try_init_ray():
import argparse
import json
from transformers.utils import strtobool
parser = argparse.ArgumentParser()
parser.add_argument('--use_ray', type=str, default='0')
parser.add_argument('--device_groups', type=str, default=None)
args, _ = parser.parse_known_args()
args.use_ray = strtobool(args.use_ray)
if args.use_ray:
RayHelper.initialize(json.loads(args.device_groups))