Update trainer.py

trainer.py

@@ -1,4 +1,18 @@
 #@title Geometric Autoregressive LM - Full Training with HF Upload + TensorBoard generated valid shakespere
+"""
+Prototype LM for geometric simplex structures.
+
+Requires the geometricvocab's SimplexFactory for valid simplex representations, or the simplex behavior will not learn.
+
+try:
+  !pip uninstall -qy geometricvocab
+except:
+  pass
+
+!pip install -q git+https://github.com/AbstractEyes/lattice_vocabulary.git
+
+License: MIT
+"""
 import torch
 import torch.nn as nn
 import torch.nn.functional as F