github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-31 10:23:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update Fireworks model names (#9085)

Browse Source
This commit is contained in:
Chenyu Zhao
2023-08-10 19:23:42 -07:00
committed by GitHub
parent a2588d6c57
commit c0acbdca1b
3 changed files with 80 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
libs/langchain/langchain/llms/fireworks.py
View File

@@ -28,9 +28,7 @@ logger = logging.getLogger(__name__)
class BaseFireworks(BaseLLM):
"""Wrapper around Fireworks large language models."""
model_id: str = Field(
"accounts/fireworks/models/fireworks-llama-v2-7b-chat", alias="model"
)
model_id: str = Field("accounts/fireworks/models/llama-v2-7b-chat", alias="model")
"""Model name to use."""
temperature: float = 0.7
"""What sampling temperature to use."""
@@ -174,10 +172,10 @@ class FireworksChat(BaseLLM):
Example:
.. code-block:: python
from langchain.llms import FireworksChat
fireworkschat = FireworksChat(model_id=""fireworks-llama-v2-13b-chat"")
fireworkschat = FireworksChat(model_id=""llama-v2-13b-chat"")
"""
model_id: str = "accounts/fireworks/models/fireworks-llama-v2-7b-chat"
model_id: str = "accounts/fireworks/models/llama-v2-7b-chat"
"""Model name to use."""
temperature: float = 0.7
"""What sampling temperature to use."""
@@ -269,7 +267,7 @@ class Fireworks(BaseFireworks):
Example:
.. code-block:: python
from langchain.llms import fireworks
llm = Fireworks(model_id="fireworks-llama-v2-13b")
llm = Fireworks(model_id="llama-v2-13b")
"""