zhongpei/Comfyui_image2prompt

torchvision error ???(IMPORT FAILED) moondream plugin

kryfo opened this issue · 1 comments

Traceback (most recent call last):
File "E:\StableDiffusion-ComfyUI\ComfyUI\nodes.py", line 1888, in load_custom_node
module_spec.loader.exec_module(module)
File "", line 883, in exec_module
File "", line 241, in call_with_frames_removed
File "E:\StableDiffusion-ComfyUI\ComfyUI\custom_nodes\Comfyui_image2prompt_init
.py", line 40, in
from .image2text import Image2Text, LoadImage2TextModel, Image2TextWithTags
File "E:\StableDiffusion-ComfyUI\ComfyUI\custom_nodes\Comfyui_image2prompt\image2text.py", line 1, in
from .moondream_model import MoondreamModel
File "E:\StableDiffusion-ComfyUI\ComfyUI\custom_nodes\Comfyui_image2prompt\moondream_model.py", line 8, in
from torchvision.transforms.v2 import (
ImportError: cannot import name 'ToImage' from 'torchvision.transforms.v2' (E:\StableDiffusion-ComfyUI\python_embeded\lib\site-packages\torchvision\transforms\v2_init_.py)

Cannot import E:\StableDiffusion-ComfyUI\ComfyUI\custom_nodes\Comfyui_image2prompt module for custom nodes: cannot import name 'ToImage' from 'torchvision.transforms.v2' (E:\StableDiffusion-ComfyUI\python_embeded\lib\site-packages\torchvision\transforms\v2_init_.py)

I'm getting the same error, has anyone managed to resolve this?