mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-06 10:34:23 +00:00
fix
This commit is contained in:
parent
8c66b7c3e9
commit
621cb93bb1
@ -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
|
||||
|
@ -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])
|
||||
|
Loading…
Reference in New Issue
Block a user