Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# neo-creative-gpt2-mini
|
| 2 |
+
|
| 3 |
+
## Overview
|
| 4 |
+
Neo-Creative is a lightweight GPT-2 model (6 layers) designed for short-form creative writing and world-building. It is optimized for speed and can be deployed on lower-resource devices.
|
| 5 |
+
|
| 6 |
+
## Model Architecture
|
| 7 |
+
- **Type:** Causal Language Model
|
| 8 |
+
- **Parameters:** ~82M parameters (Mini version)
|
| 9 |
+
- **Training:** Fine-tuned on a curated set of modern science fiction and fantasy novels.
|
| 10 |
+
|
| 11 |
+
## Intended Use
|
| 12 |
+
- Prompt generation for writers.
|
| 13 |
+
- NPC dialogue generation for video games.
|
| 14 |
+
- Creative brainstorming and poetry generation.
|
| 15 |
+
|
| 16 |
+
## Limitations
|
| 17 |
+
- Due to its small size, it may occasionally loop phrases if the temperature is too low.
|
| 18 |
+
- Does not have up-to-date factual knowledge of the real world.
|
| 19 |
+
|
| 20 |
+
## Example Code
|
| 21 |
+
```python
|
| 22 |
+
from transformers import pipeline, set_seed
|
| 23 |
+
|
| 24 |
+
generator = pipeline('text-generation', model='neo-creative-gpt2-mini')
|
| 25 |
+
set_seed(42)
|
| 26 |
+
prompt = "The stars flickered over the obsidian towers of"
|
| 27 |
+
print(generator(prompt, max_length=50, num_return_sequences=1))
|