diff --git a/moss_web_demo_streamlit.py b/moss_web_demo_streamlit.py index 17c9d3d..1b879df 100644 --- a/moss_web_demo_streamlit.py +++ b/moss_web_demo_streamlit.py @@ -1,6 +1,6 @@ import os import streamlit as st -os.environ["CUDA_VISIBLE_DEVICES"] = "7" +os.environ["CUDA_VISIBLE_DEVICES"] = "0" import time