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
if (self.instruct):
self.first_antiprompt.append(self.inp_prefix.strip())
self.first_antiprompt.append(self._tokenize(self.inp_prefix.strip()))
# primer feed
if (len(primer) > 0):