Fixes imports in train.py

fortarch · Feb 17, 2024 · 1abcd54 · 1abcd54
1 parent cc7fb34
commit 1abcd54
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -4,8 +4,8 @@
 """
 
 # feel free to use either
-from bpe_regex import Tokenizer as RegexTokenizer
-from bpe_basic import Tokenizer as BasicTokenizer
+from bpe_regex import RegexTokenizer
+from bpe_basic import BasicTokenizer
 
 # open some text and train a vocab of 512 tokens
 text = open("taylorswift.txt", "r", encoding="utf-8").read()