mirror of
https://git.datalinker.icu/kijai/ComfyUI-Hunyuan3DWrapper.git
synced 2025-12-19 01:34:26 +08:00
Update nodes.py
This commit is contained in:
parent
67a4e7a650
commit
0e18f865eb
3
nodes.py
3
nodes.py
@ -49,7 +49,7 @@ from accelerate.utils import set_module_tensor_to_device
|
|||||||
import folder_paths
|
import folder_paths
|
||||||
|
|
||||||
import comfy.model_management as mm
|
import comfy.model_management as mm
|
||||||
from comfy.utils import load_torch_file, ProgressBar
|
from comfy.utils import load_torch_file, ProgressBar, common_upscale
|
||||||
|
|
||||||
script_directory = os.path.dirname(os.path.abspath(__file__))
|
script_directory = os.path.dirname(os.path.abspath(__file__))
|
||||||
|
|
||||||
@ -232,6 +232,7 @@ class Hy3DDelightImage:
|
|||||||
delight_pipe.scheduler = self.default_scheduler
|
delight_pipe.scheduler = self.default_scheduler
|
||||||
|
|
||||||
image = image.permute(0, 3, 1, 2).to(device)
|
image = image.permute(0, 3, 1, 2).to(device)
|
||||||
|
image = common_upscale(image, width, height, "lanczos", "disabled")
|
||||||
|
|
||||||
image = delight_pipe(
|
image = delight_pipe(
|
||||||
prompt="",
|
prompt="",
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user