diff --git a/docs/swarms/models/index.md b/docs/swarms/models/index.md index 9e001eea..3ca5385e 100644 --- a/docs/swarms/models/index.md +++ b/docs/swarms/models/index.md @@ -44,7 +44,7 @@ OpenAI(api_key: str, system: str = None, console: bool = True, model: str = None ```python import asyncio -from swarms import OpenAI +from swarm_models import OpenAI chat = OpenAI(api_key="YOUR_OPENAI_API_KEY") @@ -83,7 +83,7 @@ HuggingFaceLLM(model_id: str, device: str = None, max_length: int = 20, quantize **Usage Example:** ```python -from swarms import HuggingFaceLLM +from swarm_models import HuggingFaceLLM model_id = "gpt2" hugging_face_model = HuggingFaceLLM(model_id=model_id) @@ -124,7 +124,7 @@ GooglePalm(model_name: str = "models/chat-bison-001", google_api_key: str = None **Usage Example:** ```python -from swarms import GooglePalm +from swarm_models import GooglePalm google_palm = GooglePalm() messages = [ @@ -167,7 +167,7 @@ Anthropic(model: str = "claude-2", max_tokens_to_sample: int = 256, temperature: **Usage Example:** ```python -from swarms import Anthropic +from swarm_models import Anthropic anthropic = Anthropic() prompt = "Once upon a time"