Fix stripping instruction prompt

This commit is contained in:
Mug
2023-04-04 16:20:27 +02:00
parent da5a6a7089
commit 9cde7973cc

View File

@@ -75,7 +75,7 @@ class LLaMAInteract:
# add instruction as antiprompt # add instruction as antiprompt
if (self.instruct): if (self.instruct):
self.first_antiprompt.append(self.inp_prefix.strip()) self.first_antiprompt.append(self._tokenize(self.inp_prefix.strip()))
# primer feed # primer feed
if (len(primer) > 0): if (len(primer) > 0):