This commit is contained in:
Nick
2019-09-18 23:12:25 -06:00
parent 641722fb27
commit 626acd5722
+1 -1
View File
@@ -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