Update soulchat_app.py

scutcyr · Jun 9, 2023 · cd69d25 · cd69d25
1 parent b12c369
commit cd69d25
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/soulchat_app.py b/soulchat_app.py
@@ -41,7 +41,7 @@
 # https://emoji6.com/emojiall/
 
 # 指定显卡进行推理
-os.environ['CUDA_VISIBLE_DEVICES'] = '2'
+os.environ['CUDA_VISIBLE_DEVICES'] = '0' # 默认使用0号显卡，避免Windows用户忘记修改该处
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # 下载模型并且指定路径
@@ -176,4 +176,4 @@ def get_text():
 
 if st.button("清理对话缓存"):
     st.session_state['generated'] = []
-    st.session_state['past'] = []
+    st.session_state['past'] = []