ModuleNotFoundError: No module named 'cv2'

[PATATAJEC]

Hi!

I'm curious of your Flux LayerDiffuse implementation, however I have a problem with this error:

"ModuleNotFoundError: No module named 'cv2'"

This is a whole log. Can you see how to repair?

Traceback (most recent call last): File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\demo_t2i.py", line 5, in <module> from diffusers import FluxPipeline File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\diffusers\utils\import_utils.py", line 911, in __getattr__ value = getattr(module, name) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\diffusers\utils\import_utils.py", line 911, in __getattr__ value = getattr(module, name) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\diffusers\utils\import_utils.py", line 910, in __getattr__ module = self._get_module(self._class_to_module[name]) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\diffusers\utils\import_utils.py", line 920, in _get_module return importlib.import_module("." + module_name, self.__name__) File "C:\Users\olosk\AppData\Local\Programs\Python\Python310\lib\importlib\__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\diffusers\pipelines\flux\pipeline_flux.py", line 20, in <module> from transformers import ( File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\transformers\utils\import_utils.py", line 1965, in __getattr__ value = getattr(module, name) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\transformers\utils\import_utils.py", line 1964, in __getattr__ module = self._get_module(self._class_to_module[name]) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\transformers\utils\import_utils.py", line 1976, in _get_module return importlib.import_module("." + module_name, self.__name__) File "C:\Users\olosk\AppData\Local\Programs\Python\Python310\lib\importlib\__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\transformers\models\clip\image_processing_clip.py", line 21, in <module> from ...image_processing_utils import BaseImageProcessor, BatchFeature, get_size_dict File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\transformers\image_processing_utils.py", line 22, in <module> from .image_transforms import center_crop, normalize, rescale File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\transformers\image_transforms.py", line 22, in <module> from .image_utils import ( File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\transformers\image_utils.py", line 64, in <module> from torchvision import io as torchvision_io File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torchvision\__init__.py", line 6, in <module> from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torchvision\models\__init__.py", line 2, in <module> from .convnext import * File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torchvision\models\convnext.py", line 8, in <module> from ..ops.misc import Conv2dNormActivation, Permute File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torchvision\ops\__init__.py", line 23, in <module> from .poolers import MultiScaleRoIAlign File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torchvision\ops\poolers.py", line 10, in <module> from .roi_align import roi_align File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torchvision\ops\roi_align.py", line 4, in <module> import torch._dynamo File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\_dynamo\__init__.py", line 64, in <module> torch.manual_seed = disable(torch.manual_seed) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\_dynamo\decorators.py", line 50, in disable return DisableContext()(fn) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\_dynamo\eval_frame.py", line 410, in __call__ (filename is None or trace_rules.check(fn)) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\_dynamo\trace_rules.py", line 3378, in check return check_verbose(obj, is_inlined_call).skipped File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\_dynamo\trace_rules.py", line 3361, in check_verbose rule = torch._dynamo.trace_rules.lookup_inner( File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\_dynamo\trace_rules.py", line 3442, in lookup_inner rule = get_torch_obj_rule_map().get(obj, None) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\_dynamo\trace_rules.py", line 2782, in get_torch_obj_rule_map obj = load_object(k) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\_dynamo\trace_rules.py", line 2811, in load_object val = _load_obj_from_str(x[0]) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\_dynamo\trace_rules.py", line 2795, in _load_obj_from_str return getattr(importlib.import_module(module), obj_name) File "C:\Users\olosk\AppData\Local\Programs\Python\Python310\lib\importlib\__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\nested\_internal\nested_tensor.py", line 417, in <module> values=torch.randn(3, 3, device="meta"), F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\.venv\lib\site-packages\torch\nested\_internal\nested_tensor.py:417: UserWarning: Failed to initialize NumPy: _ARRAY_API not found (Triggered internally at ..\torch\csrc\utils\tensor_numpy.cpp:84.) values=torch.randn(3, 3, device="meta"), Traceback (most recent call last): File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\demo_t2i.py", line 6, in <module> from lib_layerdiffuse.vae import TransparentVAE File "F:\Flux_LayerDiffuse\Flux-version-LayerDiffuse\lib_layerdiffuse\vae.py", line 3, in <module> import cv2 ModuleNotFoundError: No module named 'cv2'