Downloads · 30 days
9
20% of all-time downloads
rockymtn_gpt2
kmcowan/rockymtn_gpt2
rockymtn_gpt2 is a machine learning model from kmcowan. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as apache-2.0.
This is a GPT-2 model fine-tuned on novels and books of varying topics. It is designed to generate text based on the input prompt.
All-time downloads
45
Public
Parameters
124M
5.3 GB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth996 MB · 24%
At a glance
- License
- apache-2.0
- Model type
- gpt2
- Access
- Public
- Created
- Sep 22, 2024
- Updated
- Sep 30, 2024
- SHA
- 9d765386
Model card
From the Hugging Face model README
rockymtn_gpt GPT-2 Model
This is a GPT-2 model fine-tuned on novels and books of varying topics. It is designed to generate text based on the input prompt.
Model Details
- Model Name: rockymtn_gpt2
- Base Model: Custom
- Fine-tuned on: 90 books of varying topics
- Training Details: Further training to continue.
Usage
You can use this model with the transformers library:
from transformers import GPT2Tokenizer, GPT2LMHeadModel
tokenizer = GPT2Tokenizer.from_pretrained("kmcowan/rockymtn_gpt2")
model = GPT2LMHeadModel.from_pretrained("kmcowan/rockymtn_gpt2")
input_text = "The future of AI is"
input_ids = tokenizer.encode(input_text, return_tensors='pt')
outputs = model.generate(input_ids, max_length=50)
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
print(generated_text)
- Type
- gpt2
- License
- apache-2.0
- Created
- Sep 22, 2024
- Updated
- Sep 30, 2024