Fix stripping instruction prompt
This commit is contained in:
parent
da5a6a7089
commit
9cde7973cc
1 changed files with 1 additions and 1 deletions
|
@ -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):
|
||||
|
|
Loading…
Reference in a new issue