Shoriful025 commited on
Commit
413a173
·
verified ·
1 Parent(s): 410e9b8

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +27 -0
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))