Spaces:

abhisheksan
/

poetica

Running

abhisheksan commited on 6 days ago

Commit

ba9ec5c

•

1 Parent(s): bcc47be

rollback

Files changed (1) hide show

main.py CHANGED Viewed

@@ -23,8 +23,8 @@ BATCH_SIZE = 4
 CACHE_SIZE = 1024
 MODEL_CONFIG = GPT2Config(
-    n_positions=512,
-    n_ctx=512,
     n_embd=384,
     n_layer=6,
     n_head=6,

 CACHE_SIZE = 1024
 MODEL_CONFIG = GPT2Config(
+    n_positions=128,
+    n_ctx=128,
     n_embd=384,
     n_layer=6,
     n_head=6,