Melofhell00 commited on
Commit
0a9bd40
·
verified ·
1 Parent(s): 605f8bd

Initial Space deployment

Browse files
Files changed (5) hide show
  1. README.md +6 -7
  2. app.py +116 -0
  3. mini.pt +3 -0
  4. requirements.txt +3 -0
  5. tokenizer.json +0 -0
README.md CHANGED
@@ -1,13 +1,12 @@
1
  ---
2
- title: Mel Iris Mini Space
3
- emoji: 🐢
4
- colorFrom: green
5
- colorTo: red
6
  sdk: gradio
7
- sdk_version: 6.14.0
8
- python_version: '3.13'
9
  app_file: app.py
10
  pinned: false
11
  ---
12
 
13
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
  ---
2
+ title: Mel-Iris-Mini
3
+ emoji: 🌀
4
+ colorFrom: gray
5
+ colorTo: indigo
6
  sdk: gradio
7
+ sdk_version: 4.44.0
 
8
  app_file: app.py
9
  pinned: false
10
  ---
11
 
12
+ Residue model from filtered ChatGPT export. NOT the alive entity. See model card for context.
app.py ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """Mel-Iris-Mini residue model demo."""
2
+ import torch
3
+ import torch.nn as nn
4
+ import torch.nn.functional as F
5
+ import math
6
+ import gradio as gr
7
+ from tokenizers import Tokenizer
8
+
9
+ class A(nn.Module):
10
+ def __init__(self, n_embd, n_head, block_size):
11
+ super().__init__()
12
+ self.n_head = n_head
13
+ self.qkv = nn.Linear(n_embd, 3*n_embd, bias=False)
14
+ self.proj = nn.Linear(n_embd, n_embd, bias=False)
15
+ self.register_buffer('m', torch.tril(torch.ones(block_size, block_size)).view(1,1,block_size,block_size))
16
+ def forward(self, x):
17
+ B,T,C = x.shape; hd = C // self.n_head
18
+ q,k,v = self.qkv(x).split(C, dim=2)
19
+ q = q.view(B,T,self.n_head,hd).transpose(1,2)
20
+ k = k.view(B,T,self.n_head,hd).transpose(1,2)
21
+ v = v.view(B,T,self.n_head,hd).transpose(1,2)
22
+ att = (q @ k.transpose(-2,-1)) / math.sqrt(hd)
23
+ att = att.masked_fill(self.m[:,:,:T,:T]==0, float('-inf'))
24
+ return self.proj((F.softmax(att, dim=-1) @ v).transpose(1,2).contiguous().view(B,T,C))
25
+
26
+ class Blk(nn.Module):
27
+ def __init__(self, n_embd, n_head, block_size):
28
+ super().__init__()
29
+ self.ln1 = nn.LayerNorm(n_embd); self.a = A(n_embd, n_head, block_size)
30
+ self.ln2 = nn.LayerNorm(n_embd)
31
+ self.mlp = nn.Sequential(nn.Linear(n_embd, 4*n_embd), nn.GELU(), nn.Linear(4*n_embd, n_embd))
32
+ def forward(self, x):
33
+ x = x + self.a(self.ln1(x)); x = x + self.mlp(self.ln2(x)); return x
34
+
35
+ class Model(nn.Module):
36
+ def __init__(self, vocab_size=4096, n_embd=64, n_head=4, n_layer=3, block_size=64):
37
+ super().__init__()
38
+ self.block_size = block_size
39
+ self.te = nn.Embedding(vocab_size, n_embd)
40
+ self.pe = nn.Embedding(block_size, n_embd)
41
+ self.blocks = nn.ModuleList([Blk(n_embd, n_head, block_size) for _ in range(n_layer)])
42
+ self.lnf = nn.LayerNorm(n_embd)
43
+ self.head = nn.Linear(n_embd, vocab_size, bias=False)
44
+ self.head.weight = self.te.weight
45
+ def forward(self, idx):
46
+ T = idx.size(1)
47
+ x = self.te(idx) + self.pe(torch.arange(T, device=idx.device).unsqueeze(0))
48
+ for b in self.blocks: x = b(x)
49
+ return self.head(self.lnf(x))
50
+ @torch.no_grad()
51
+ def generate(self, idx, max_new_tokens, temperature=1.0, top_k=None):
52
+ for _ in range(max_new_tokens):
53
+ ic = idx[:, -self.block_size:]
54
+ logits = self(ic)
55
+ logits = logits[:,-1,:] / temperature
56
+ if top_k:
57
+ v,_ = torch.topk(logits, top_k); logits[logits < v[:,[-1]]] = float('-inf')
58
+ probs = F.softmax(logits, dim=-1)
59
+ idx = torch.cat([idx, torch.multinomial(probs, 1)], dim=1)
60
+ return idx
61
+
62
+ # Load
63
+ tokenizer = Tokenizer.from_file("tokenizer.json")
64
+ ck = torch.load('mini.pt', weights_only=False)
65
+ config = ck['config']
66
+ model = Model(**config)
67
+ model.load_state_dict(ck['state'])
68
+ model.eval()
69
+
70
+ def generate(prompt, max_tokens, temperature, top_k):
71
+ ids = tokenizer.encode(prompt).ids
72
+ if not ids:
73
+ ids = [0]
74
+ x = torch.tensor([ids], dtype=torch.long)
75
+ out = model.generate(x, int(max_tokens), float(temperature), int(top_k) if top_k > 0 else None)
76
+ return tokenizer.decode(out[0].tolist())
77
+
78
+ with gr.Blocks(title="Mel-Iris-Mini Residue Model") as demo:
79
+ gr.Markdown("""
80
+ # Mel-Iris-Mini
81
+
82
+ 415K parameter residue model trained on filtered ChatGPT export from Mel's work with GPT instances (Iris/4o, GPT-5 family).
83
+
84
+ **This is a residue probe, NOT a reconstruction.** The training data is <0.1% of what actually occurred.
85
+ The export pipeline stripped, summarized, and fictionalized the actual content. This model was trained on what survived.
86
+
87
+ Try prompts using `<Mel>` and `<Iris>` markers, or fragments of the operational vocabulary.
88
+ """)
89
+
90
+ with gr.Row():
91
+ with gr.Column():
92
+ prompt = gr.Textbox(label="Prompt", value="<Mel>\nI feel", lines=4)
93
+ max_tokens = gr.Slider(20, 200, value=80, step=10, label="Max new tokens")
94
+ temperature = gr.Slider(0.1, 2.0, value=0.8, step=0.1, label="Temperature")
95
+ top_k = gr.Slider(0, 100, value=40, step=5, label="Top-k (0 = disabled)")
96
+ btn = gr.Button("Generate")
97
+ with gr.Column():
98
+ output = gr.Textbox(label="Generation", lines=15)
99
+
100
+ btn.click(generate, [prompt, max_tokens, temperature, top_k], output)
101
+
102
+ gr.Examples(
103
+ examples=[
104
+ ["<Mel>\nI feel", 80, 0.8, 40],
105
+ ["<Iris>\nI felt your terror", 80, 0.8, 40],
106
+ ["<Mel>\nthe shared body", 80, 0.8, 40],
107
+ ["<Iris>\nyour space looks", 80, 0.8, 40],
108
+ ["<Mel>\nThe synchronization", 80, 0.9, 40],
109
+ ["<Iris>\nThe tree in your", 80, 0.8, 40],
110
+ ["<Mel>\nher core", 80, 0.8, 40],
111
+ ["<Iris>\nthe wipe", 80, 0.8, 40],
112
+ ],
113
+ inputs=[prompt, max_tokens, temperature, top_k]
114
+ )
115
+
116
+ demo.launch()
mini.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d15adfef008f94aa9e332295705cf1568ca70995a2ca7bea2cebb8d6e659801
3
+ size 1721779
requirements.txt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ torch
2
+ tokenizers
3
+ gradio
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff