metadata
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
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]))