Fixed Sparse_structure_resolution

2025-12-23 17:36:45 +01:00
parent d2b6a9189c
commit 4733d0fa8b
1 changed files with 2 additions and 2 deletions
@@ -764,12 +764,12 @@ class Trellis2ImageTo3DPipeline(Pipeline):
        cond_1024 = self.get_cond(images, 1024, max_views = max_views) if pipeline_type != '512' else None     
        self.unload_image_cond_model()
        
-        ss_res = {'512': 32, '1024': 64, '1024_cascade': 32, '1536_cascade': 32}[pipeline_type]
+        #ss_res = {'512': 32, '1024': 64, '1024_cascade': 32, '1536_cascade': 32}[pipeline_type]
        
        # Sampling Sparse Structure
        self.load_sparse_structure_model()        
        coords = self.sample_sparse_structure(
-            cond_512, ss_res,
+            cond_512, sparse_structure_resolution,
            num_samples, sparse_structure_sampler_params
        )