File size: 609 Bytes
f7b7243 |
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 |
---
license: apache-2.0
---
# Model Card
This model was trained using custom training code.
## Model Details
- **Architecture**: Decoder-only transformer
- **Parameters**: See config.json for details
- **Training**: Custom training loop
## Usage
```python
from transformers import AutoTokenizer, AutoModelForCausalLM
tokenizer = AutoTokenizer.from_pretrained("omkaark/test1")
model = AutoModelForCausalLM.from_pretrained("omkaark/test1")
# Generate text
inputs = tokenizer("Hello, world!", return_tensors="pt")
outputs = model.generate(**inputs, max_length=50)
print(tokenizer.decode(outputs[0]))
```
|