OpenBMB/VisCPM

VisCPM-Paint推理报错

Maxhyl opened this issue · 0 comments

环境:
python310
torch 2.2.1
cuda 12.3

运行代码:
from VisCPM import VisCPMPaint
painter = VisCPMPaint('./models/VisCPM-Paint', image_safety_checker=True, prompt_safety_checker=True, add_ranker=True)

默认对输入的文本和输出的图片进行安全检查,默认开启重排序

image = painter.generate('人闲桂花落,月静春山空') # 对应上图第一行第二张图片
image.save('./test.png')

报错如下:
Traceback (most recent call last):
File "/home/edcuser/models/VisCPM-main/predict-paint.py", line 2, in
from VisCPM import VisCPMPaint
File "/home/edcuser/models/VisCPM-main/VisCPM/init.py", line 4, in
from VisCPM.viscpm_chat import VisCPMChat
File "/home/edcuser/models/VisCPM-main/VisCPM/viscpm_chat.py", line 7, in
from timm.models import create_model
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/timm/init.py", line 2, in
from .models import create_model, list_models, is_model, list_modules, model_entrypoint,
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/timm/models/init.py", line 1, in
from .byoanet import *
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/timm/models/byoanet.py", line 15, in
from timm.data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/timm/data/init.py", line 5, in
from .dataset import ImageDataset, IterableImageDataset, AugMixDataset
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/timm/data/dataset.py", line 12, in
from .parsers import create_parser
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/timm/data/parsers/init.py", line 1, in
from .parser_factory import create_parser
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/timm/data/parsers/parser_factory.py", line 3, in
from .parser_image_folder import ParserImageFolder
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/timm/data/parsers/parser_image_folder.py", line 10, in
from timm.utils.misc import natural_key
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/timm/utils/init.py", line 13, in
from .summary import update_summary, get_outdir
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/timm/utils/summary.py", line 9, in
import wandb
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/init.py", line 27, in
from wandb import sdk as wandb_sdk
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/sdk/init.py", line 28, in
from .wandb_init import _attach, init
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/sdk/wandb_init.py", line 32, in
from .backend.backend import Backend
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/sdk/backend/backend.py", line 19, in
from wandb.sdk.internal.internal import wandb_internal
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/sdk/internal/internal.py", line 32, in
from . import context, handler, internal_util, sender, writer
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/sdk/internal/sender.py", line 32, in
from wandb.filesync.dir_watcher import DirWatcher
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/filesync/dir_watcher.py", line 22, in
wd_polling = util.vendor_import("wandb_watchdog.observers.polling")
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/util.py", line 193, in vendor_import
module = import_module(name)
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/importlib/init.py", line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/init.py", line 63, in
from .inotify import InotifyObserver as Observer
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify.py", line 74, in
from .inotify_buffer import InotifyBuffer
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify_buffer.py", line 20, in
from wandb_watchdog.observers.inotify_c import Inotify
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify_c.py", line 62, in
libc = _load_libc()
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/wandb/vendor/watchdog_0_9_0/wandb_watchdog/observers/inotify_c.py", line 42, in _load_libc
return ctypes.CDLL(libc_path)
File "/home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/ctypes/init.py", line 374, in init
self._handle = _dlopen(self._name, mode)
OSError: /home/edcuser/.conda/envs/cuda123_torch221/lib/python3.10/site-packages/amp_C.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN2at4_ops5zeros4callEN3c108ArrayRefINS2_6SymIntEEENS2_8optionalINS2_10ScalarTypeEEENS6_INS2_6LayoutEEENS6_INS2_6DeviceEEENS6_IbEE
请问这个问题怎么解决?