diff options
author | Kyomotoi <[email protected]> | 2022-06-25 22:38:05 +0800 |
---|---|---|
committer | Kyomotoi <[email protected]> | 2022-06-25 22:38:05 +0800 |
commit | 7920895ab697fd8c45e21387e6344252671ef069 (patch) | |
tree | dbab8c9a9ca6aa911d2a1d206c57a0e30d4ffe3f /ATRI/plugins/setu | |
parent | 37df264d7e9acb624767bdb654eed01d49137aa3 (diff) | |
download | ATRI-7920895ab697fd8c45e21387e6344252671ef069.tar.gz ATRI-7920895ab697fd8c45e21387e6344252671ef069.tar.bz2 ATRI-7920895ab697fd8c45e21387e6344252671ef069.zip |
🚑️ 修复启动时因可能的报错导致强制退出
Diffstat (limited to 'ATRI/plugins/setu')
-rw-r--r-- | ATRI/plugins/setu/tf_dealer.py | 6 |
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("装载模型失败") |