This commit is contained in:
ishaan-jaff 2023-09-19 11:05:48 -07:00
parent 85862c1066
commit 5b2f952035

View file

@ -1,6 +1,12 @@
# Petals # Petals
Petals: https://github.com/bigscience-workshop/petals Petals: https://github.com/bigscience-workshop/petals
## Pre-Requisites
Ensure you have `petals` installed
```shell
pip install git+https://github.com/bigscience-workshop/petals
```
## Usage ## Usage
Ensure you add `petals/` as a prefix for all petals LLMs. This sets the custom_llm_provider to petals Ensure you add `petals/` as a prefix for all petals LLMs. This sets the custom_llm_provider to petals
@ -16,12 +22,10 @@ print(response)
``` ```
### Model Details ### Model Details
| Model Name | Function Call | | Model Name | Function Call |
|------------------|--------------------------------------------| |------------------|--------------------------------------------|
| StableBeluga | ```python | StableBeluga | `completion('petals/petals-team/StableBeluga2', messages)` |
| completion(
| model="petals/petals-team/StableBeluga2",
| messages=messages
| )
| ``` |