noanabeshima · noanabeshima · Jul 5, 2024 · Jul 5, 2024 · Jul 7, 2024 · Jul 7, 2024
diff --git a/.gitignore b/.gitignore
@@ -5,6 +5,8 @@ testing.ipynb
 dist/
 *.gitignore
 mlp_map_test/
-attn_test/
 simulation.py
-test.pt
+attn_test/
+test.pt
+mlp/
+
diff --git a/README.md b/README.md
@@ -8,7 +8,7 @@ This library is in an alpha state, it probably has some bugs. Please let me know
 
 
 ```
-from tiny_model import TinyModel, tokenizer
+from tinymodel import TinyModel, tokenizer
 
 lm = TinyModel()
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,10 +1,10 @@
 [tool.poetry]
 name = "tinymodel"
-version = "0.1.2-1"
+version = "0.1.2-2"
 description = "A small TinyStories LM with SAEs and transcoders"
 authors = ["Noa Nabeshima <[email protected]>"]
 readme = "README.md"
-packages = [{include = "tiny_model"}]
+packages = [{include = "tinymodel"}]
 
 [tool.poetry.dependencies]
 python = "^3.11"
@@ -19,7 +19,7 @@ datasets = "^2.20.0"
 
 
 [[tool.poetry.source]]
-name = "tiny_model"
+name = "tinymodel"
 url = "https://github.com/noanabeshima/tiny_model"
 priority = "explicit"
 

diff --git a/tiny_model/lm.py b/tiny_model/lm.py
diff --git a/tiny_model/sparse_mlp.py b/tiny_model/sparse_mlp.py
diff --git a/tiny_model/__init__.py → tinymodel/__init__.py b/tiny_model/__init__.py → tinymodel/__init__.py
@@ -1,3 +1,3 @@
-from .lm import TinyModel
+from .lm import TinyModel, parse_mlp_tag
 from .sparse_mlp import SparseMLP
-from .tokenization.tokenization import dec, enc, tok_see, tokenizer, raw_toks, toks
+from .tokenization.tokenization import dec, enc, tok_see, tokenizer, raw_toks, pretty_toks, multi_tok_words, words
-Original file line number
+Diff line change
@@ Expand Up @@
     ```
-    from tiny_model import TinyModel, tokenizer
+    from tinymodel import TinyModel, tokenizer
     lm = TinyModel()
@@ Expand Down @@