Skip to content

Commit 4833929

Browse files
committed
fix docs
1 parent f9ab29b commit 4833929

File tree

4 files changed

+5
-5
lines changed

4 files changed

+5
-5
lines changed

docs/source/package_reference/models.mdx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@
77

88
## Accelerate and Transformers Models
99
### TransformersModel
10-
[[autodoc]] models.transformers.base_model.TransformersModelConfig
11-
[[autodoc]] models.transformers.base_model.TransformersModel
10+
[[autodoc]] models.transformers.transformers_model.TransformersModelConfig
11+
[[autodoc]] models.transformers.transformers_model.TransformersModel
1212

1313
### AdapterModel
1414
[[autodoc]] models.transformers.adapter_model.AdapterModelConfig

src/lighteval/main_accelerate.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def accelerate( # noqa C901
4444
model_args: Annotated[
4545
str,
4646
Argument(
47-
help="Model arguments in the form key1=value1,key2=value2,... or path to yaml config file (see examples/model_configs/base_model.yaml)"
47+
help="Model arguments in the form key1=value1,key2=value2,... or path to yaml config file (see examples/model_configs/transformers_model.yaml)"
4848
),
4949
],
5050
tasks: Annotated[str, Argument(help="Comma-separated list of tasks to evaluate on.")],

src/lighteval/main_endpoint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def openai(
4646
model_args: Annotated[
4747
str,
4848
Argument(
49-
help="Model name as a string (has to be available through the openai API) or path to yaml config file (see examples/model_configs/base_model.yaml)"
49+
help="Model name as a string (has to be available through the openai API) or path to yaml config file (see examples/model_configs/transformers_model.yaml)"
5050
),
5151
],
5252
tasks: Annotated[str, Argument(help="Comma-separated list of tasks to evaluate on.")],

src/lighteval/main_vllm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def vllm(
4040
model_args: Annotated[
4141
str,
4242
Argument(
43-
help="Model arguments in the form key1=value1,key2=value2,... or path to yaml config file (see examples/model_configs/base_model.yaml)"
43+
help="Model arguments in the form key1=value1,key2=value2,... or path to yaml config file (see examples/model_configs/transformers_model.yaml)"
4444
),
4545
],
4646
tasks: Annotated[str, Argument(help="Comma-separated list of tasks to evaluate on.")],

0 commit comments

Comments
 (0)