fix: guanaco output

This commit is contained in:
csunny
2023-06-04 20:15:27 +08:00
parent 7adc30d6b9
commit f7fe66b5e5

View File

@@ -57,7 +57,7 @@ class BaseOutputParser(ABC):
output = data["text"][skip_echo_len:].strip()
elif "guanaco" in CFG.LLM_MODEL:
# output = data["text"][skip_echo_len + 14:].replace("<s>", "").strip()
output = data["text"][skip_echo_len:].replace("<s>", "").strip()
output = data["text"][skip_echo_len + 2:].replace("<s>", "").strip()
else:
output = data["text"].strip()