ExternalVendorCode/lollms
1
0
Fork 0
mirror of https://github.com/ParisNeo/lollms.git synced 2024-12-18 20:27:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source 
Added a "Response Time" feature which tells the user the amount of time elapsed during text generation in milliseconds
This commit is contained in:
Flameguy677 2023-08-07 21:09:30 +05:30 committed by GitHub
parent 6b650343b5
commit 9a7f3278ef
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lollms/apps/console/__init__.py
View File

@ -11,6 +11,7 @@ from typing import Callable
from pathlib import Path
import argparse
import yaml
import time
import sys
class Conversation(LollmsApplication):
@ -178,6 +179,7 @@ Participating personalities:
ump = self.config.discussion_prompt_separator +self.config.user_name+": " if self.config.use_user_name_in_discussions else self.personality.user_message_prefix
if self.config.use_user_name_in_discussions:
prompt = input(f"{ASCIIColors.color_green}{self.config.user_name}: {ASCIIColors.color_reset}")
t0 = time.time() #Time at start of request
else:
prompt = input(f"{ASCIIColors.color_green}You: {ASCIIColors.color_reset}")
if prompt == "exit":
@ -271,6 +273,8 @@ Participating personalities:
self.log(full_discussion)
t1 = time.time() # Time at end of defense
print(f"{ASCIIColors.color_cyan}Response Time: {ASCIIColors.color_reset}",str(int((t1-t0)*1000)),"ms\n") # Total time elapsed since t0 in ms
except KeyboardInterrupt:
print("Keyboard interrupt detected.\nBye")
break
@ -320,4 +324,4 @@ def main():
if __name__ == "__main__":
main()
main()