nodes/lerignoux-ComfyUI-Stable3DGen
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixing trellis model folder in case multiple extra path are provided.

Browse Source
This commit is contained in:
Erignoux Laurent
2025-08-05 13:46:52 +08:00
parent 786c305cbc
commit c1aa4b0e90
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
stable_3d.py
+1 -2
View File
@@ -126,8 +126,7 @@ class Stable3DLoadModels:
# download dinov2 feature detection model and library to ~/.cache/torch/hub/
torch.hub.load('facebookresearch/dinov2', 'dinov2_vitl14_reg', pretrained=True)
trellis_folder = folder_paths.get_folder_paths("trellis")[0]
hi3dgen_pipeline = Hi3DGenPipeline.from_pretrained(os.path.join(trellis_folder, trellis_model))
hi3dgen_pipeline = Hi3DGenPipeline.from_pretrained(os.path.join(self.models_path, trellis_model))
hi3dgen_pipeline.cuda()
self.load_birefnet_model(hi3dgen_pipeline, birefnet_model)