YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

KING RIZAW v0.1

A small language model trained on Shakespeare's works, generating text in a Shakespearean style with creative flair.

Model Description

  • Model Type: Transformer-based language model
  • Training Data: Shakespeare's works
  • Model Size: ~160K parameters
  • Use Cases: Creative text generation, Shakespeare-style text continuations
  • Language: English

Usage

# Example code to use the model
import torch
from transformers import AutoTokenizer, AutoModelForCausalLM

model_name = "Antonioguerra/KING_RIZAW_V0.1"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)

prompt = "To be or not to be,"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(inputs["input_ids"], max_length=50, temperature=0.8)
result = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(result)

Training

This model was trained on a collection of Shakespeare's works using a transformer architecture with self-attention mechanisms. The training process utilized both standard training approaches and enhanced techniques to improve the quality of generated text.

Training Curves

Parameters

  • Temperature: Controls creativity - higher values produce more varied outputs
  • Max Tokens: Maximum length of generated text (1 token ≈ 4 characters)

License

This model is released under an open license for research and creative purposes.

Authors

  • Antonio Guerra
  • Sayb Saad Ahmed
Downloads last month
12
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 1 Ask for provider support