clean up

kyegomez · Oct 9, 2023 · 5e218b1 · 5e218b1
1 parent 76df402
commit 5e218b1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -16,13 +16,15 @@ GPT for proteins sequences
 
 # Usage
 ```python
-from progen_torch import ProGen
+import torch
+from progen.model import ProGen
 
 x = torch.randint(0, 100, (1, 1024))
-model = ProGenBase(num_tokens=100, dim=512, seq_len=1024, depth=6)
+model = ProGen(num_tokens=100, dim=512, seq_len=1024, depth=6)
 outputs = model(x)
 print(outputs)
 
+
 ```
 
 # Architecture

diff --git a/example.py b/example.py
@@ -4,4 +4,4 @@
 x = torch.randint(0, 100, (1, 1024))
 model = ProGen(num_tokens=100, dim=512, seq_len=1024, depth=6)
 outputs = model(x)
-print(outputs)
+print(outputs)