RYDE-WORK/Langchain-Chatchat
1
0
Fork 0
mirror of https://github.com/RYDE-WORK/Langchain-Chatchat.git synced 2026-02-06 23:15:53 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update xinference manager ui

Browse Source
This commit is contained in:
liunux4odoo 2024-03-06 13:03:57 +08:00
parent 49bc5b54a4
commit f0f31167e9
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
model_loaders/xinference_manager.py
View File

@ -208,10 +208,10 @@ st.divider()
st.markdown(text) st.markdown(text)
st.divider() st.divider()
cols = st.columns([3, 1]) local_path = st.text_input("本地模型绝对路径：")
local_path = cols[0].text_input("本地模型绝对路径：") cols = st.columns(3)
if cols[1].button("设置模型缓存"): if cols[0].button("设置模型缓存"):
if os.path.isabs(local_path) and os.path.isdir(local_path): if os.path.isabs(local_path) and os.path.isdir(local_path):
cur_spec.model_uri = local_path cur_spec.model_uri = local_path
if os.path.isdir(cache_dir): if os.path.isdir(cache_dir):
@ -238,7 +238,7 @@ if cols[1].button("删除模型缓存"):
if os.path.isdir(cache_dir): if os.path.isdir(cache_dir):
os.rmdir(cache_dir) os.rmdir(cache_dir)
if cols[1].button("注册为自定义模型"): if cols[2].button("注册为自定义模型"):
if os.path.isabs(local_path) and os.path.isdir(local_path): if os.path.isabs(local_path) and os.path.isdir(local_path):
cur_spec.model_uri = local_path cur_spec.model_uri = local_path
cur_spec.model_revision = None cur_spec.model_revision = None