39 lines
936 B
Markdown
39 lines
936 B
Markdown
---
|
|
library_name: transformers
|
|
tags:
|
|
- conversational
|
|
- text-generation
|
|
- text-generation-inference
|
|
license: apache-2.0
|
|
language:
|
|
- en
|
|
---
|
|
|
|
# KryzeLLM
|
|
|
|
A fine-tuned conversational AI assistant developed by **Ravindra Vala**.
|
|
|
|
## Model Details
|
|
|
|
- **Developed by:** Ravindra Vala
|
|
- **Model type:** Causal Language Model
|
|
- **Language(s):** English
|
|
- **License:** Apache 2.0
|
|
|
|
## How to Get Started
|
|
|
|
```python
|
|
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
import torch
|
|
|
|
model = AutoModelForCausalLM.from_pretrained("ravindravala/kryzeLLM")
|
|
tokenizer = AutoTokenizer.from_pretrained("ravindravala/kryzeLLM")
|
|
|
|
messages = [{"role": "user", "content": "Who are you?"}]
|
|
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
|
inputs = tokenizer(prompt, return_tensors="pt")
|
|
|
|
output = model.generate(**inputs, max_new_tokens=100)
|
|
print(tokenizer.decode(output[0], skip_special_tokens=True))
|
|
```
|