Owl-v1 is a lightweight 10.8 M parameter language model built from scratch for learning and integration purposes.
Character name / scene opener prompts.
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("IstishadAlamTishad/Owl-v1", trust_remote_code=True)
5model = AutoModelForCausalLM.from_pretrained("IstishadAlamTishad/Owl-v1", trust_remote_code=True)
6model.eval()
7
8ids = tokenizer.encode("HAMLET:\n", return_tensors="pt")
9with torch.no_grad():
10 out = model.generate(ids, max_new_tokens=200, do_sample=True, temperature=0.8, top_k=40)
11print(tokenizer.decode(out[0].tolist()))