Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,106 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- geometry-dash
|
| 5 |
+
- level-generation
|
| 6 |
+
- gguf
|
| 7 |
+
- tinyllama
|
| 8 |
+
base_model: TinyLlama/TinyLlama-1.1B-Chat-v1.0
|
| 9 |
+
pipeline_tag: text-generation
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# EditorAI
|
| 13 |
+
|
| 14 |
+
The main EditorAI model — a fine-tuned TinyLlama-1.1B-Chat that generates Geometry Dash levels as JSON with blocks, spikes, platforms, triggers, groups, color channels, and more.
|
| 15 |
+
|
| 16 |
+
Part of the [EditorAI](https://github.com/Entity12208/EditorAI) project — an AI-powered level generator mod for Geometry Dash.
|
| 17 |
+
|
| 18 |
+
## About EditorAI
|
| 19 |
+
|
| 20 |
+
[EditorAI](https://github.com/Entity12208/EditorAI) is a [Geode](https://geode-sdk.org) mod for Geometry Dash that lets you describe a level in plain text and have AI build it in the editor. It supports 8 AI providers (Gemini, Claude, OpenAI, Mistral, HuggingFace, Ollama, LM Studio, llama.cpp) and features blueprint preview, feedback learning, 15+ trigger types, and an in-game settings UI.
|
| 21 |
+
|
| 22 |
+
## Model Details
|
| 23 |
+
|
| 24 |
+
- **Base model:** TinyLlama-1.1B-Chat-v1.0
|
| 25 |
+
- **Training:** QLoRA (4-bit, rank 8) on 2368 examples (368 expert-crafted + 2000 synthetic), 2 epochs
|
| 26 |
+
- **Features:** Blocks, spikes, platforms, color triggers, move triggers, alpha triggers, rotate triggers, toggle triggers, pulse triggers, spawn triggers, stop triggers, speed portals, groups, color channels
|
| 27 |
+
- **GGUF quantization:** q4_k_m (637 MB)
|
| 28 |
+
- **Tested:** 8/8 tests passed, generates all trigger types
|
| 29 |
+
|
| 30 |
+
## Files
|
| 31 |
+
|
| 32 |
+
| File | Size | Description |
|
| 33 |
+
|------|------|-------------|
|
| 34 |
+
| `model.safetensors` | 2.1 GB | Merged fp16 model weights |
|
| 35 |
+
| `editorai-latest.gguf` | 637 MB | Quantized GGUF (q4_k_m) for llama.cpp / LM Studio |
|
| 36 |
+
| `config.json` | — | Model architecture config |
|
| 37 |
+
| `tokenizer.json` | — | Tokenizer |
|
| 38 |
+
|
| 39 |
+
## Setup
|
| 40 |
+
|
| 41 |
+
This model uses the **Zephyr/ChatML** chat template and works best with the following system prompt:
|
| 42 |
+
|
| 43 |
+
```
|
| 44 |
+
You are a Geometry Dash level designer. Return ONLY valid JSON with an analysis string and objects array. Each object needs type, x, y. Y >= 0. X uses 10 units per grid cell.
|
| 45 |
+
```
|
| 46 |
+
|
| 47 |
+
> **Recommended:** Use the Ollama version (`entity12208/editorai:latest`) which has the system prompt and template pre-configured.
|
| 48 |
+
|
| 49 |
+
## Usage with llama.cpp
|
| 50 |
+
|
| 51 |
+
```bash
|
| 52 |
+
wget https://huggingface.co/EditorAI-Geode/EditorAI/resolve/main/editorai-latest.gguf
|
| 53 |
+
|
| 54 |
+
llama-server -m editorai-latest.gguf --port 8080 --chat-template chatml
|
| 55 |
+
|
| 56 |
+
# In the EditorAI mod: set provider to "llama-cpp", URL to http://localhost:8080
|
| 57 |
+
```
|
| 58 |
+
|
| 59 |
+
## Usage with LM Studio
|
| 60 |
+
|
| 61 |
+
1. Download `editorai-latest.gguf` from this repo
|
| 62 |
+
2. Load it in LM Studio, set **Prompt Format** to **ChatML**
|
| 63 |
+
3. Set the **System Prompt** to the prompt above
|
| 64 |
+
4. Start the server
|
| 65 |
+
5. In the EditorAI mod: set provider to "lm-studio", URL to `http://localhost:1234`
|
| 66 |
+
|
| 67 |
+
## Usage with Ollama (recommended)
|
| 68 |
+
|
| 69 |
+
```bash
|
| 70 |
+
ollama pull entity12208/editorai:latest
|
| 71 |
+
```
|
| 72 |
+
|
| 73 |
+
In the EditorAI mod: set provider to "ollama" and select `entity12208/editorai:latest`.
|
| 74 |
+
|
| 75 |
+
## Output Format
|
| 76 |
+
|
| 77 |
+
```json
|
| 78 |
+
{
|
| 79 |
+
"analysis": "A medium modern level with color transitions and moving platforms.",
|
| 80 |
+
"objects": [
|
| 81 |
+
{"type": "block_black_gradient_square", "x": 0, "y": 0, "color_channel": 10},
|
| 82 |
+
{"type": "spike_black_gradient_spike", "x": 50, "y": 0},
|
| 83 |
+
{"type": "color_trigger", "x": 80, "y": 0, "color_channel": 1, "color": "#0066FF", "duration": 1.5},
|
| 84 |
+
{"type": "move_trigger", "x": 90, "y": 0, "target_group": 1, "move_x": 0, "move_y": 20, "duration": 1.0, "easing": 1},
|
| 85 |
+
{"type": "end_trigger", "x": 400, "y": 0}
|
| 86 |
+
]
|
| 87 |
+
}
|
| 88 |
+
```
|
| 89 |
+
|
| 90 |
+
## Model Comparison
|
| 91 |
+
|
| 92 |
+
| Model | Size | Triggers | Quality | Speed |
|
| 93 |
+
|-------|------|----------|---------|-------|
|
| 94 |
+
| **editorai:latest** (this) | 637 MB | All types | Best | ~15-30s |
|
| 95 |
+
| editorai:mini | 379 MB | Color, move | Good | ~10-20s |
|
| 96 |
+
|
| 97 |
+
## Links
|
| 98 |
+
|
| 99 |
+
- **Mod:** [github.com/Entity12208/EditorAI](https://github.com/Entity12208/EditorAI)
|
| 100 |
+
- **Ollama:** [ollama.com/entity12208/editorai](https://ollama.com/entity12208/editorai)
|
| 101 |
+
- **Mini model:** [huggingface.co/EditorAI-Geode/editorai-mini](https://huggingface.co/EditorAI-Geode/editorai-mini)
|
| 102 |
+
- **Discord:** [discord.gg/5hwCqMUYNj](https://discord.gg/5hwCqMUYNj)
|
| 103 |
+
|
| 104 |
+
## License
|
| 105 |
+
|
| 106 |
+
Apache 2.0
|