From a8268fa29dc02cac5f89ef97b29771f85f48e867 Mon Sep 17 00:00:00 2001 From: Kye Date: Tue, 24 Oct 2023 13:35:42 -0400 Subject: [PATCH] gpt2 Former-commit-id: e519dc1d06ebaa435b38940ee89291f9333f0615 --- docs/swarms/models/huggingface.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/swarms/models/huggingface.md b/docs/swarms/models/huggingface.md index 841d7a39..e429f080 100644 --- a/docs/swarms/models/huggingface.md +++ b/docs/swarms/models/huggingface.md @@ -93,7 +93,7 @@ Here are three ways to use the `HuggingfaceLLM` class: #### Example 1: Basic Usage ```python -from your_module import HuggingfaceLLM +from swarms.models import HuggingfaceLLM # Initialize the HuggingfaceLLM instance with a model ID model_id = "gpt2-small" @@ -108,7 +108,7 @@ print(generated_text) #### Example 2: Custom Configuration ```python -from your_module import HuggingfaceLLM +from swarms.models import HuggingfaceLLM # Initialize with custom configuration custom_config = { @@ -127,7 +127,7 @@ print(generated_text) #### Example 3: Distributed Processing ```python -from your_module import HuggingfaceLLM +from swarms.models import HuggingfaceLLM # Initialize for distributed processing inference = HuggingfaceLLM(model_id="gpt2-medium", distributed=True)