summaryrefslogtreecommitdiff
path: root/ATRI/plugins/setu
diff options
context:
space:
mode:
authorKyomotoi <[email protected]>2022-06-25 22:38:05 +0800
committerKyomotoi <[email protected]>2022-06-25 22:38:05 +0800
commit7920895ab697fd8c45e21387e6344252671ef069 (patch)
treedbab8c9a9ca6aa911d2a1d206c57a0e30d4ffe3f /ATRI/plugins/setu
parent37df264d7e9acb624767bdb654eed01d49137aa3 (diff)
downloadATRI-7920895ab697fd8c45e21387e6344252671ef069.tar.gz
ATRI-7920895ab697fd8c45e21387e6344252671ef069.tar.bz2
ATRI-7920895ab697fd8c45e21387e6344252671ef069.zip
🚑️ 修复启动时因可能的报错导致强制退出
Diffstat (limited to 'ATRI/plugins/setu')
-rw-r--r--ATRI/plugins/setu/tf_dealer.py6
1 files changed, 3 insertions, 3 deletions
diff --git a/ATRI/plugins/setu/tf_dealer.py b/ATRI/plugins/setu/tf_dealer.py
index f56c579..1e51f54 100644
--- a/ATRI/plugins/setu/tf_dealer.py
+++ b/ATRI/plugins/setu/tf_dealer.py
@@ -102,10 +102,10 @@ async def init_module():
path = SETU_PATH / file_name
if not path.is_file():
log.warning("缺少模型文件,装载中")
- data = await request.get(MODULE_URL)
try:
+ data = await request.get(MODULE_URL)
with open(path, "wb") as w:
w.write(data.read())
log.info("模型装载完成")
- except WriteFileError:
- raise WriteFileError("NSFW TF module init failed!")
+ except Exception:
+ log.error("装载模型失败")