bakaEC/OliveSensorAPI
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update web_demo_Llama3.py

Browse Source
This commit is contained in:
HongCheng 2024-05-03 03:02:14 +09:00
parent b83270bb13
commit 8d6571be46
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
web_demo_Llama3.py
View File

@ -214,7 +214,7 @@ def combine_history(prompt):
def main(): def main():
torch.cuda.empty_cache() # torch.cuda.empty_cache()
print("load model begin.") print("load model begin.")
model, tokenizer = load_model() model, tokenizer = load_model()
print("load model end.") print("load model end.")
@ -222,7 +222,7 @@ def main():
user_avator = "assets/user.png" user_avator = "assets/user.png"
robot_avator = "assets/EmoLLM.png" robot_avator = "assets/EmoLLM.png"
st.title("EmoLLM Llama3心理咨询室V2.0") st.title("EmoLLM Llama3心理咨询室V3.0")
generation_config = prepare_generation_config() generation_config = prepare_generation_config()
@ -268,5 +268,5 @@ def main():
torch.cuda.empty_cache() torch.cuda.empty_cache()
if __name__ == "__main__": if __name__ == '__main__':
main() main()