Update app.py

app.py

@@ -16,7 +16,7 @@ config = GPTConfig(
     n_layer=12,
     n_head=12,
     n_embd=768,
-    n_kv_head=6,
+    n_kv_head=12,
     sequence_len=1024,
     # dropout=0.1,
 )