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
|
# 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):
|
||||||
|
|
Loading…
Reference in a new issue