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

rollback Fix Blackwell -> does not work

Browse Source
This commit is contained in:
Bruno Fargnoli
2026-04-02 10:38:56 +02:00
parent 531e53c3f3
commit a5242473f6
2 changed files with 3 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+1 -1
View File
@@ -14,7 +14,7 @@
| Date | Description |
| --- | --- |
| **2026-04-01** | Added node "Voxel to Mesh"<br>It replaces Remeshing to make watertight mesh<br>Added "Blackwell Fix" in "Load Model" node |
| **2026-04-01** | Added node "Voxel to Mesh"<br>It replaces Remeshing to make watertight mesh |
| **2026-03-21** | Added node "Projection HighPoly to LowPoly"<br>Added node "Render MultiView" |
| **2026-03-17** | Added Inpainting Choice NS and TELEA |
| **2026-03-14** | Added Experimental node "Projection MultiView Texturing"<br>Check in example_workflows folder |
nodes.py
+2 -10
View File
@@ -329,7 +329,6 @@ class Trellis2LoadModel:
"device": (["cpu","cuda"],{"default":"cuda"}),
"low_vram": ("BOOLEAN",{"default":True}),
"keep_models_loaded": ("BOOLEAN", {"default":True}),
"fix_blackwell": ("BOOLEAN",{"default":False}),
"conv_backend": (["spconv","torchsparse","flex_gemm"],{"default":"flex_gemm"}),
"sparse_backend": (["xformers","flash_attn"],{"default":"flash_attn"}),
},
@@ -341,15 +340,7 @@ class Trellis2LoadModel:
CATEGORY = "Trellis2Wrapper"
OUTPUT_NODE = True
def process(self, modelname, backend, device, low_vram, keep_models_loaded, fix_blackwell, conv_backend, sparse_backend):
if fix_blackwell:
from .blackwell_fix import patch_all
patch_all()
print('conv_backend is forced to spconv')
sparseconfig.set_conv_backend('spconv')
else:
sparseconfig.set_conv_backend(conv_backend)
def process(self, modelname, backend, device, low_vram, keep_models_loaded, conv_backend, sparse_backend):
os.environ['OPENCV_IO_ENABLE_OPENEXR'] = '1'
os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True" # Can save GPU memory
#os.environ["FLEX_GEMM_AUTOTUNE_CACHE_PATH"] = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'autotune_cache.json')
@@ -358,6 +349,7 @@ class Trellis2LoadModel:
config.set_backend(backend)
sparseconfig.set_attn_backend(sparse_backend)
sparseconfig.set_conv_backend(conv_backend)
reset_cuda()