Skip to content

VisCPM-Paint推理报错 #47

@Maxhyl

Description

@Maxhyl

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions