From 26c0a67851dfb3c4f0536d563d3bf2c56416640e Mon Sep 17 00:00:00 2001 From: Kye Date: Mon, 23 Oct 2023 18:51:42 -0400 Subject: [PATCH] contributing md --- README.md | 2 +- swarms/models/llava.py | 8 ++++---- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 0cfe49e8..fa70e755 100644 --- a/README.md +++ b/README.md @@ -163,7 +163,7 @@ For documentation, go here, [swarms.apac.ai](https://swarms.apac.ai) ----- ## Contribute -We're always looking for contributors to help us improve and expand this project. If you're interested, please check out our [Contributing Guidelines](C0NTRIBUTING.md). +We're always looking for contributors to help us improve and expand this project. If you're interested, please check out our [Contributing Guidelines](CONTRIBUTING.md). ### Optimization Priorities diff --git a/swarms/models/llava.py b/swarms/models/llava.py index 67c0e4a7..6f8019bc 100644 --- a/swarms/models/llava.py +++ b/swarms/models/llava.py @@ -16,16 +16,17 @@ class MultiModalLlava: top_k: The top k value for sampling repetition_penalty: The repetition penalty for sampling device_map: The device map to use - + Methods: __call__: Call the model chat: Interactive chat in terminal - + Example: >>> from swarms.models.llava import LlavaModel >>> model = LlavaModel(device="cpu") >>> model("Hello, I am a robot.") """ + def __init__( self, model_name_or_path="TheBloke/llava-v1.5-13B-GPTQ", @@ -37,7 +38,7 @@ class MultiModalLlava: top_p=0.95, top_k=40, repetition_penalty=1.1, - device_map: str = "auto" + device_map: str = "auto", ): self.device = device self.model = AutoModelForCausalLM.from_pretrained( @@ -76,4 +77,3 @@ class MultiModalLlava: break response = self(user_input) print(f"Model: {response}") -