Skip to content

Commit

Permalink
Update README.md to include inference example
Browse files Browse the repository at this point in the history
Signed-off-by: Andriy Mulyar <[email protected]>
  • Loading branch information
AndriyMulyar authored Jun 28, 2023
1 parent a67f813 commit 390994e
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions gpt4all-api/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -43,3 +43,29 @@ localhost:80/docs
```

This documentation should match the OpenAI OpenAPI spec located at https://github.com/openai/openai-openapi/blob/master/openapi.yaml


#### Running inference
```python
import openai
openai.api_base = "http://localhost:4891/v1"

openai.api_key = "not needed for a local LLM"


def test_completion():
model = "gpt4all-j-v1.3-groovy"
prompt = "Who is Michael Jordan?"
response = openai.Completion.create(
model=model,
prompt=prompt,
max_tokens=50,
temperature=0.28,
top_p=0.95,
n=1,
echo=True,
stream=False
)
assert len(response['choices'][0]['text']) > len(prompt)
print(response)
```

0 comments on commit 390994e

Please sign in to comment.