Skip to content

Commit 256e8b5

Browse files
docs: add ollama to readme (#113)
1 parent 6518f2c commit 256e8b5

File tree

1 file changed

+11
-10
lines changed

1 file changed

+11
-10
lines changed

README.md

Lines changed: 11 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,6 @@ pip install plexe
3737

3838
You can use plexe as a Python library to build and train machine learning models:
3939

40-
#### B. Library API
41-
4240
```python
4341
import plexe
4442

@@ -106,9 +104,9 @@ Plexe supports distributed model training and evaluation with Ray for faster par
106104

107105
```python
108106
from plexe import Model
109-
from plexe.config import config
110107

111108
# Optional: Configure Ray cluster address if using remote Ray
109+
# from plexe import config
112110
# config.ray.address = "ray://10.1.2.3:10001"
113111

114112
model = Model(
@@ -140,13 +138,18 @@ model.build(provider="openai/gpt-4o-mini") # Schema inferred automatically
140138
```
141139

142140
### 2.6. 🌐 Multi-Provider Support
143-
Use your preferred LLM provider:
141+
Use your preferred LLM provider, for example:
144142
```python
145-
model.build(provider="openai/gpt-4o-mini") # OpenAI
146-
model.build(provider="anthropic/claude-3-opus") # Anthropic
147-
model.build(provider="google/gemini-1.5-pro") # Google
143+
model.build(provider="openai/gpt-4o-mini") # OpenAI
144+
model.build(provider="anthropic/claude-3-opus") # Anthropic
145+
model.build(provider="ollama/llama2") # Ollama
146+
model.build(provider="huggingface/meta-llama/...") # Hugging Face
148147
```
149-
See [LiteLLM providers](https://docs.litellm.ai/docs/providers) for available providers.
148+
See [LiteLLM providers](https://docs.litellm.ai/docs/providers) for instructions and available providers.
149+
150+
> [!NOTE]
151+
> Plexe *should* work with most LiteLLM providers, but we actively test only with `openai/*` and `anthropic/*`
152+
> models. If you encounter issues with other providers, please let us know.
150153
151154

152155
## 3. Installation
@@ -170,8 +173,6 @@ See [LiteLLM providers](https://docs.litellm.ai/docs/providers) for environment
170173
## 4. Documentation
171174
For full documentation, visit [docs.plexe.ai](https://docs.plexe.ai).
172175

173-
174-
175176
## 5. Contributing
176177
See [CONTRIBUTING.md](CONTRIBUTING.md) for guidelines. Join our [Discord](https://discord.gg/SefZDepGMv) to connect with the team.
177178

0 commit comments

Comments
 (0)