Skip to content

Commit

Permalink
Add model name param metadata (#3425)
Browse files Browse the repository at this point in the history
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
  • Loading branch information
connor-mccorm and pre-commit-ci[bot] committed May 28, 2023
1 parent 22e302c commit 2bada30
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 0 deletions.
3 changes: 3 additions & 0 deletions ludwig/schema/metadata/configs/llm.yaml
@@ -1,3 +1,6 @@
model_name:
ui_display_name: Model Name
expected_impact: 3
generation:
temperature:
ui_display_name: Temperature
Expand Down
2 changes: 2 additions & 0 deletions ludwig/schema/model_types/llm.py
Expand Up @@ -14,6 +14,7 @@
)
from ludwig.schema.generation import LLMGenerationConfig, LLMGenerationConfigField
from ludwig.schema.hyperopt import HyperoptConfig, HyperoptField
from ludwig.schema.metadata import LLM_METADATA
from ludwig.schema.model_types.base import ModelConfig, register_model_type
from ludwig.schema.preprocessing import PreprocessingConfig, PreprocessingField
from ludwig.schema.prompt import PromptConfig, PromptConfigField
Expand Down Expand Up @@ -48,6 +49,7 @@ def __post_init__(self):
"model name or a path to a local directory containing a valid "
"HuggingFace model."
),
parameter_metadata=LLM_METADATA["model_name"],
)

input_features: FeatureCollection[BaseInputFeatureConfig] = LLMInputFeatureSelection().get_list_field()
Expand Down

0 comments on commit 2bada30

Please sign in to comment.