github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-08 03:24:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements/requirements.txt
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

1
tests/test_booster/test_plugin/test_gemini_plugin.py
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])