gpt2-witcher-books

GPT-2 Fine-tuned on The Witcher Books

A GPT-2 Medium model fine-tuned on text from The Witcher book series by Andrzej Sapkowski for creative text generation in a dark fantasy style.

Model Description

This model is part of the Deepstory project, which combines Natural Language Generation, Text-to-Speech, and animation technologies to create interactive storytelling experiences.

The model has been fine-tuned on The Witcher novels to generate text that captures the narrative style, vocabulary, and themes of the original books.

Model Architecture

This model is based on GPT-2 Medium architecture.

Usage

With Transformers Library

from transformers import GPT2Tokenizer, GPT2LMHeadModel
import torch

# Load model and tokenizer
tokenizer = GPT2Tokenizer.from_pretrained("thetobysiu/gpt2-witcher-books")
model = GPT2LMHeadModel.from_pretrained("thetobysiu/gpt2-witcher-books")

# Generate text
prompt = "The witcher drew his silver sword and"
input_ids = tokenizer.encode(prompt, return_tensors='pt')

output = model.generate(
    input_ids=input_ids,
    max_length=150,
    temperature=0.9,
    top_p=0.95,
    top_k=50,
    do_sample=True,
    num_return_sequences=1
)

generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
print(generated_text)

Generation Parameters

For best results, we recommend the following generation parameters:

Training Data

The model was fine-tuned on text from The Witcher book series, including:

• The Last Wish
• Sword of Destiny
• Blood of Elves
• Time of Contempt
• Baptism of Fire
• The Tower of the Swallow
• The Lady of the Lake
• Season of Storms

Training Procedure

The model was fine-tuned using the Hugging Face Transformers library starting from the pre-trained GPT-2 Medium checkpoint.

Intended Use

This model is intended for:

• Creative writing assistance in fantasy genre
• Interactive storytelling applications
• Fan fiction generation
• Research in language model fine-tuning
• Educational purposes

Limitations

• The model may generate content that doesn't perfectly match the original author's style
• Context length is limited to 1024 tokens
• May occasionally generate repetitive or incoherent text
• The training data is based on copyrighted material

Ethical Considerations

• This model is trained on copyrighted material and should be used for personal/research purposes
• Generated content should not be presented as original work by the book author
• Users should be aware that generated text may contain mature themes consistent with the source material

Citation

If you use this model, please cite the Deepstory project:

@misc{deepstory,
  author = {Siu King Wai},
  title = {Deepstory},
  year = {2020},
  publisher = {GitHub},
  url = {https://github.com/thetobysiu/deepstory}
}

License

This model is released under the MIT License. Please note that The Witcher series is the intellectual property of Andrzej Sapkowski and CD Projekt.

Acknowledgments

• OpenAI for the original GPT-2 model
• Hugging Face for the Transformers library
• Andrzej Sapkowski for The Witcher book series