Merge pull request #402 from MrForExample/dev

Fix ComfyUI-Manager Install error

Author: MrForExample

Gen_3D_Modules/InstantMesh/utils/infer_util.py

@@ -1,6 +1,5 @@
 import os
 import imageio
-import rembg
 import torch
 import numpy as np
 import PIL.Image
@@ -13,6 +12,7 @@ def remove_background(image: PIL.Image.Image,
     force: bool = False,
     **rembg_kwargs,
 ) -> PIL.Image.Image:
+    import rembg
     do_remove = True
     if image.mode == "RGBA" and image.getextrema()[3][0] < 255:
         do_remove = False

Gen_3D_Modules/StableFast3D/sf3d/utils.py

@@ -1,7 +1,6 @@
 from typing import Any
 
 import numpy as np
-import rembg
 import torch
 from PIL import Image
 
@@ -41,6 +40,7 @@ def remove_background(
     force: bool = False,
     **rembg_kwargs,
 ) -> Image:
+    import rembg
     do_remove = True
     if image.mode == "RGBA" and image.getextrema()[3][0] < 255:
         do_remove = False

Gen_3D_Modules/TriplaneGaussian/utils/ops.py

@@ -5,7 +5,6 @@
 import torch.nn.functional as F
 from torch.autograd import Function
 from torch.amp import custom_bwd, custom_fwd
-from pytorch3d import io
 from pytorch3d.renderer import (
     PointsRasterizationSettings, 
     PointsRasterizer)

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-3d-pack"
 description = "Make ComfyUI generates 3D assets as good & convenient as it generates image/video!"
-version = "0.1.3.alpha"
+version = "0.1.3"
 license = { file = "LICENSE" }
 dynamic = ["dependencies"]
 [tool.setuptools.dynamic]

requirements.txt

@@ -31,6 +31,7 @@ peft
 
 # for process images & videos
 rembg
+onnxruntime
 imageio
 imageio-ffmpeg
 matplotlib