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