From 0d3eb451f25884c810459edb8664d042e21f0577 Mon Sep 17 00:00:00 2001 From: Refound <137168144+Refound-445@users.noreply.github.com> Date: Thu, 28 Nov 2024 00:12:39 +0800 Subject: [PATCH] Update common.py --- .../nonebot_plugin_nailongremove/model/utils/common.py | 1 + 1 file changed, 1 insertion(+) diff --git a/packages/nonebot-plugin-nailongremove-base/nonebot_plugin_nailongremove/model/utils/common.py b/packages/nonebot-plugin-nailongremove-base/nonebot_plugin_nailongremove/model/utils/common.py index 37382f3..2569ba1 100644 --- a/packages/nonebot-plugin-nailongremove-base/nonebot_plugin_nailongremove/model/utils/common.py +++ b/packages/nonebot-plugin-nailongremove-base/nonebot_plugin_nailongremove/model/utils/common.py @@ -29,6 +29,7 @@ if config.nailong_similarity_on: from nonebot import logger from torch import nn from torchvision import transforms + import sklearn transform = transforms.Compose( [