OpenBMB / VisCPM

[ICLR'24 spotlight] Chinese and English Multimodal Large Model Series (Chat and Paint) | 基于CPM基础模型的中英双语多模态大模型系列
1.06k stars 93 forks source link

VisCPM-Paint推理报错 #47

Open Maxhyl opened 2 months ago

Maxhyl commented 2 months ago

环境: 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 请问这个问题怎么解决?