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

Update web_demo-aiwei.py

Browse Source
This commit is contained in:
xzw 2024-03-15 21:12:24 +08:00 committed by GitHub
parent ccfa75c493
commit 5485b4c124
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
web_demo-aiwei.py
View File

@ -210,6 +210,8 @@ def combine_history(prompt):
def main(): def main():
st.markdown("我在这里，准备好倾听你的心声了。", unsafe_allow_html=True)
# torch.cuda.empty_cache() # torch.cuda.empty_cache()
print("load model begin.") print("load model begin.")
model, tokenizer = load_model() model, tokenizer = load_model()