From c1aa4b0e90d2deeb0d938512662a14c8f29322eb Mon Sep 17 00:00:00 2001 From: Erignoux Laurent Date: Tue, 5 Aug 2025 13:46:12 +0800 Subject: [PATCH] Fixing trellis model folder in case multiple extra path are provided. --- stable_3d.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/stable_3d.py b/stable_3d.py index bb78259..e9ca680 100644 --- a/stable_3d.py +++ b/stable_3d.py @@ -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)