fjxddy/author-stylegan

Author StyleGAN Models

This repository contains fine-tuned GPT-2 models trained to generate text in the style of various authors.

Available Authors

• Agatha Christie
• Alexandre Dumas
• Arthur Conan Doyle
• Charles Dickens
• Charlotte Brontë
• F. Scott Fitzgerald
• García Márquez
• Herman Melville
• Jane Austen
• Mark Twain

Model Details

• Model type: GPT-2 with StyleGAN training
• Training data: Custom dataset of authors' works
• Training procedure: StyleGAN training with BERT discriminator
• Authors: Your Name

Usage

from transformers import AutoModelForCausalLM, AutoTokenizer

# 选择作者
author = "Agatha_Christie"  # 或其他作者

# 加载模型
model = AutoModelForCausalLM.from_pretrained(
    f"{username}/{model_name}/generators/{author}"
)
tokenizer = AutoTokenizer.from_pretrained(
    f"{username}/{model_name}/generators/{author}"
)