Apply suggestions from code review

This commit is contained in:
Raphael Mitsch 2023-07-11 12:12:05 +02:00 committed by GitHub
parent 1e28ce8c82
commit 14f6a07b61
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1352,12 +1352,12 @@ python -m pip install "accelerate>=0.16.0,<1.0"
> ```ini > ```ini
> [components.llm.model] > [components.llm.model]
> @llm_models = "spacy.OpenLLaMA.v1" > @llm_models = "spacy.OpenLLaMA.v1"
> name = "open_llama_3b_350bt_preview" > name = "open_llama_3b"
> ``` > ```
| Argument | Description | | Argument | Description |
| ------------- | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------ | | ------------- | ------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------ |
| `name` | The name of a OpenLLaMA model that is supported (e. g. "open_llama_3b_350bt_preview"). ~~Literal["open_llama_3b_350bt_preview", "open_llama_3b_600bt_preview", "open_llama_7b_400bt_preview", "open_llama_7b_600bt_preview"]~~ | | `name` | The name of a OpenLLaMA model that is supported. ~~Literal["open_llama_3b", "open_llama_7b", "open_llama_7b_v2", "open_llama_13b"]~~ |
| `config_init` | Further configuration passed on to the construction of the model with `transformers.AutoModelForCausalLM.from_pretrained()`. Defaults to `{}`. ~~Dict[str, Any]~~ | | `config_init` | Further configuration passed on to the construction of the model with `transformers.AutoModelForCausalLM.from_pretrained()`. Defaults to `{}`. ~~Dict[str, Any]~~ |
| `config_run` | Further configuration used during model inference. Defaults to `{}`. ~~Dict[str, Any]~~ | | `config_run` | Further configuration used during model inference. Defaults to `{}`. ~~Dict[str, Any]~~ |