From 626acd5722eca54a9fe956f19f4a79c6b9d4a419 Mon Sep 17 00:00:00 2001
From: Nick <nickwalton00@gmail.com>
Date: Wed, 18 Sep 2019 23:12:25 -0600
Subject: [PATCH] update

---
 generator/ctrl/ctrl_generator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generator/ctrl/ctrl_generator.py b/generator/ctrl/ctrl_generator.py
index 4bad39f..4fc6705 100644
--- a/generator/ctrl/ctrl_generator.py
+++ b/generator/ctrl/ctrl_generator.py
@@ -126,7 +126,7 @@ class CTRLGenerator():
 
         # almost there, we now take the user prompt and tokenize with BPE
         # load BPE codes
-        self.bpe = fastBPE('codes', 'vocab')
+        self.bpe = fastBPE.fastBPE('codes', 'vocab')
 
         self.temperature = 0
         self.nucleusprob = 0