This commit is contained in:
flybird11111 2025-03-31 16:16:15 +08:00
parent 8c66b7c3e9
commit 621cb93bb1
2 changed files with 1 additions and 2 deletions

View File

@ -16,7 +16,7 @@ ray
sentencepiece
google
protobuf
transformers==4.50.0
transformers==4.39.3
peft>=0.7.1,<=0.13.2
bitsandbytes>=0.39.0
rpyc==6.0.0

View File

@ -67,7 +67,6 @@ def run_fn(init_method, model_fn, data_gen_fn, output_transform_fn, zero_size, t
# TODO(ver217): CI does not support lazy now
# @parameterize('init_method', ['lazy', 'none', 'colo'])
@parameterize("subset", [COMMON_MODELS] if IS_FAST_TEST else ["torchvision", "transformers", "diffusers"])
@parameterize("init_method", ["none"])
@parameterize("zero_size", [2])