"Add HWC3 casting. Fix #443

pyproject.toml

@@ -1,9 +1,10 @@
 [project]
 name = "comfyui_controlnet_aux"
 description = "Plug-and-play ComfyUI node sets for making ControlNet hint images"
-version = "1.0.4-alpha.5"
-license = "LICENSE"
-dependencies = ["torch", "importlib_metadata", "huggingface_hub", "scipy", "opencv-python>=4.7.0.72", "filelock", "numpy", "Pillow", "einops", "torchvision", "pyyaml", "scikit-image", "python-dateutil", "mediapipe", "svglib", "fvcore", "yapf", "omegaconf", "ftfy", "addict", "yacs", "trimesh[easy]", "albumentations", "scikit-learn"]
+
+version = "1.0.4-alpha.7"
+license = { file = "LICENSE.txt" }
+dependencies = ["torch", "importlib_metadata", "huggingface_hub", "scipy", "opencv-python>=4.7.0.72", "filelock", "numpy", "Pillow", "einops", "torchvision", "pyyaml", "scikit-image", "python-dateutil", "mediapipe", "svglib", "fvcore", "yapf", "omegaconf", "ftfy", "addict", "yacs", "trimesh[easy]", "albumentations", "scikit-learn", "matplotlib"]
 
 [project.urls]
 Repository = "https://github.com/Fannovel16/comfyui_controlnet_aux"

requirements.txt

@@ -22,3 +22,4 @@ yacs
 trimesh[easy]
 albumentations
 scikit-learn
+matplotlib

src/custom_controlnet_aux/metric3d/mono/model/backbones/ConvNeXt.py

@@ -1,8 +1,8 @@
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from timm.models.layers import trunc_normal_, DropPath
-from timm.models.registry import register_model
+from custom_timm.models.layers import trunc_normal_, DropPath
+from custom_timm.models.registry import register_model
 
 class Block(nn.Module):
     r""" ConvNeXt Block. There are two equivalent implementations: