File size: 796 Bytes
6f3ada8 d480edd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 |
# Custom Urdu LLM
This is a custom transformer-based Large Language Model for Urdu.
## Model Details
- **Architecture:** Transformer (GPT-based)
- **Framework:** PyTorch
- **Tokenizer:** SentencePiece
- **Hyperparameters:**
- Vocabulary Size: 20,000
- Embedding Size: 768
- Attention Heads: 12
- Layers: 12
- Dropout: 0.2
## Usage
```python
from transformers import AutoModel, AutoTokenizer
model = AutoModel.from_pretrained("AliMuhammad73/testing-model")
tokenizer = AutoTokenizer.from_pretrained("AliMuhammad73/testing-model")
prompt = <prompt in urdu>
inputs = tokenizer(prompt, return_tensors="pt")
output = model.generate(inputs.input_ids, max_new_tokens=tokens_to_generate)
print(tokenizer.decode(output[0]))
```
---
license: apache-2.0
---
|