ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2025-04-08 11:24:14 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

upgraded

Browse Source
This commit is contained in:
Saifeddine ALOUI 2024-04-19 22:30:45 +02:00
parent f4a9bbec32
commit d1f6f07942
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lollms/app.py
View File

@ -303,7 +303,7 @@ class LollmsApplication(LoLLMsCom):
generation_infos["generated_text"] += chunk
antiprompt = self.personality.detect_antiprompt(generation_infos["generated_text"])
if antiprompt:
ASCIIColors.warning(f"\nDetected hallucination with antiprompt: {antiprompt}")
ASCIIColors.warning(f"\n{antiprompt} detected. Stopping generation")
generation_infos["generated_text"] = self.remove_text_from_string(generation_infos["generated_text"],antiprompt)
return False
else:

2
lollms/personality.py
View File

@ -699,7 +699,7 @@ class AIPersonality:
antiprompt = self.detect_antiprompt(bot_says)
if antiprompt:
self.bot_says = self.remove_text_from_string(bot_says,antiprompt)
ASCIIColors.warning(f"\nDetected hallucination with antiprompt: {antiprompt}")
ASCIIColors.warning(f"\n{antiprompt} detected. Stopping generation")
return False
else:
if callback:

4
lollms/server/endpoints/lollms_generator.py
View File

@ -441,7 +441,7 @@ async def v1_completion(request: CompletionGenerationRequest):
output["text"] += chunk
antiprompt = detect_antiprompt(output["text"])
if antiprompt:
ASCIIColors.warning(f"\nDetected hallucination with antiprompt: {antiprompt}")
ASCIIColors.warning(f"\n{antiprompt} detected. Stopping generation")
output["text"] = remove_text_from_string(output["text"],antiprompt)
return False
else:
@ -462,7 +462,7 @@ async def v1_completion(request: CompletionGenerationRequest):
output["text"] += chunk
antiprompt = detect_antiprompt(output["text"])
if antiprompt:
ASCIIColors.warning(f"\nDetected hallucination with antiprompt: {antiprompt}")
ASCIIColors.warning(f"\n{antiprompt} detected. Stopping generation")
output["text"] = remove_text_from_string(output["text"],antiprompt)
return False
else: