[Deprecation][2/N] Replace --task with --runner and --convert (#21470)
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk> Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -69,10 +69,7 @@ async def test_guided_logits_processor_black_box(backend: str, is_local: bool,
|
||||
|
||||
config = ModelConfig(
|
||||
MODEL_NAME,
|
||||
task="generate",
|
||||
tokenizer=MODEL_NAME,
|
||||
tokenizer_mode="auto",
|
||||
trust_remote_code=False,
|
||||
runner="generate",
|
||||
seed=0,
|
||||
dtype="bfloat16",
|
||||
)
|
||||
@@ -113,10 +110,7 @@ async def test_guided_logits_processor_with_reasoning(
|
||||
|
||||
config = ModelConfig(
|
||||
REASONING_MODEL_NAME,
|
||||
task="generate",
|
||||
tokenizer=REASONING_MODEL_NAME,
|
||||
tokenizer_mode="auto",
|
||||
trust_remote_code=False,
|
||||
runner="generate",
|
||||
seed=0,
|
||||
dtype="bfloat16",
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user