From 5608651bfabf8dad589debbd78b08e1cb2aba583 Mon Sep 17 00:00:00 2001 From: Kye Date: Tue, 24 Oct 2023 13:35:42 -0400 Subject: [PATCH] gpt2 Former-commit-id: 62f80ff949fa9b343769ab8b8b0d2114a07bc375 --- 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)