Skip to content

Commit cd69d25

Browse files
authored
Update soulchat_app.py
1 parent b12c369 commit cd69d25

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

soulchat_app.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
# https://emoji6.com/emojiall/
4242

4343
# 指定显卡进行推理
44-
os.environ['CUDA_VISIBLE_DEVICES'] = '2'
44+
os.environ['CUDA_VISIBLE_DEVICES'] = '0' # 默认使用0号显卡,避免Windows用户忘记修改该处
4545
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
4646

4747
# 下载模型并且指定路径
@@ -176,4 +176,4 @@ def get_text():
176176

177177
if st.button("清理对话缓存"):
178178
st.session_state['generated'] = []
179-
st.session_state['past'] = []
179+
st.session_state['past'] = []

0 commit comments

Comments
 (0)