File tree Expand file tree Collapse file tree 4 files changed +5
-5
lines changed
llama-index-integrations/llms/llama-index-llms-cerebras
llama_index/llms/cerebras Expand file tree Collapse file tree 4 files changed +5
-5
lines changed Original file line number Diff line number Diff line change 103
103
" \" Enter your Cerebras API key: \"\n " ,
104
104
" )\n " ,
105
105
" \n " ,
106
- " llm = Cerebras(model=\" llama3.1 -70b\" , api_key=os.environ[\" CEREBRAS_API_KEY\" ])"
106
+ " llm = Cerebras(model=\" llama-3.3 -70b\" , api_key=os.environ[\" CEREBRAS_API_KEY\" ])"
107
107
]
108
108
},
109
109
{
Original file line number Diff line number Diff line change @@ -44,7 +44,7 @@ import os
44
44
from llama_index.core.llms import ChatMessage
45
45
from llama_index.llms.cerebras import Cerebras
46
46
47
- llm = Cerebras(model = " llama3.1 -70b" , api_key = os.environ[" CEREBRAS_API_KEY" ])
47
+ llm = Cerebras(model = " llama-3.3 -70b" , api_key = os.environ[" CEREBRAS_API_KEY" ])
48
48
49
49
messages = [
50
50
ChatMessage(
@@ -63,7 +63,7 @@ import os
63
63
64
64
from llama_index.llms.cerebras import Cerebras
65
65
66
- llm = Cerebras(model = " llama3.1 -70b" , api_key = os.environ[" CEREBRAS_API_KEY" ])
66
+ llm = Cerebras(model = " llama-3.3 -70b" , api_key = os.environ[" CEREBRAS_API_KEY" ])
67
67
68
68
response = llm.stream_complete(" What is Generative AI?" )
69
69
for r in response:
Original file line number Diff line number Diff line change @@ -15,7 +15,7 @@ class Cerebras(OpenAILike):
15
15
from llama_index.llms.cerebras import Cerebras
16
16
17
17
# Set up the Cerebras class with the required model and API key
18
- llm = Cerebras(model="llama3.1 -70b", api_key="your_api_key")
18
+ llm = Cerebras(model="llama-3.3 -70b", api_key="your_api_key")
19
19
20
20
# Call the complete method with a query
21
21
response = llm.complete("Why is fast inference important?")
Original file line number Diff line number Diff line change @@ -30,7 +30,7 @@ license = "MIT"
30
30
name = " llama-index-llms-cerebras"
31
31
packages = [{include = " llama_index/" }]
32
32
readme = " README.md"
33
- version = " 0.2.1 "
33
+ version = " 0.2.2 "
34
34
35
35
[tool .poetry .dependencies ]
36
36
python = " >=3.9,<4.0"
You can’t perform that action at this time.
0 commit comments