mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2026-01-13 09:30:34 +08:00
[v1] add sft (#9752)
This commit is contained in:
@@ -19,7 +19,7 @@ from llamafactory.v1.core.model_engine import ModelEngine
|
||||
|
||||
|
||||
def test_init_on_meta():
|
||||
_, model_args, *_ = get_args(
|
||||
model_args, *_ = get_args(
|
||||
dict(
|
||||
model="llamafactory/tiny-random-qwen3",
|
||||
init_config={"name": "init_on_meta"},
|
||||
@@ -30,7 +30,7 @@ def test_init_on_meta():
|
||||
|
||||
|
||||
def test_init_on_rank0():
|
||||
_, model_args, *_ = get_args(
|
||||
model_args, *_ = get_args(
|
||||
dict(
|
||||
model="llamafactory/tiny-random-qwen3",
|
||||
init_config={"name": "init_on_rank0"},
|
||||
@@ -44,7 +44,7 @@ def test_init_on_rank0():
|
||||
|
||||
|
||||
def test_init_on_default():
|
||||
_, model_args, *_ = get_args(
|
||||
model_args, *_ = get_args(
|
||||
dict(
|
||||
model="llamafactory/tiny-random-qwen3",
|
||||
init_config={"name": "init_on_default"},
|
||||
|
||||
Reference in New Issue
Block a user