|
--- |
|
language: |
|
- en |
|
license: apache-2.0 |
|
tags: |
|
- text-generation-inference |
|
- transformers |
|
- unsloth |
|
- gemma |
|
- trl |
|
base_model: unsloth/gemma-7b-bnb-4bit |
|
--- |
|
|
|
# Uploaded model |
|
|
|
- **Developed by:** Mollel |
|
- **License:** apache-2.0 |
|
- **Finetuned from model :** unsloth/gemma-7b-bnb-4bit |
|
|
|
```python3 |
|
|
|
from llama_index.llms.huggingface import HuggingFaceLLM |
|
llm = HuggingFaceLLM( |
|
context_window=4096, |
|
max_new_tokens=256, |
|
generate_kwargs={"temperature": 0.7, "do_sample": False}, |
|
tokenizer_name="Mollel/Swahili_Gemma", |
|
model_name="Mollel/Swahili_Gemma", |
|
device_map="auto", |
|
stopping_ids=[50278, 50279, 50277, 1, 0], |
|
tokenizer_kwargs={"max_length": 4096}, |
|
model_kwargs={"torch_dtype": torch.float16} |
|
) |
|
|
|
``` |
|
|
|
|
|
|
