train

kyegomez · Jul 2, 2023 · 10c6488 · 10c6488
1 parent 32233b3
commit 10c6488
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -27,13 +27,16 @@ First `pip install Andromeda-llm` then
 
 ```python
 import torch
-from Andromeda import Andromeda
+from Andromeda import Andromeda, Train
 
 
 x = torch.randint(0, 20000, (1, 1024))
 
 Andromea(x)
 
+# or train
+
+Train()
 
 ```