Rtx 5090 - TRELLIS.2 설치

1. Custom Node 설치

- ComfyUI-TRELLIS2-Pozzetti

- ComfyUI-Trellis2-visualbruno 

 

2. ComfyUI\custom_nodes\ComfyUI-Trellis2-visualbruno\wheels 폴더의 Windows \ Torch 280 의 Whl들을 자신의 버전에 맞는 whl을 선택 ex) 312 해서 cumesh, flex_gemm, nvdiffrast, nvdiffrec_render, o_voxel 을 ComfyUI\python_embeded 폴더 환경에 모두 설치해 주어야 한다.

 

3. Trellis2ImageToShape  오류 해결 방법

No operator found for `memory_efficient_attention_forward` with inputs:

query : shape=(1, 4096, 12, 128) (torch.bfloat16)

key : shape=(1, 4096, 12, 128) (torch.bfloat16)

value : shape=(1, 4096, 12, 128) (torch.bfloat16)

attn_bias : <class 'NoneType'>

p : 0.0

`fa3F@2.8.3-133-gde1584b` is not supported because:

requires device with capability <= (9, 0) but your GPU has capability (12, 0) (too new)

requires device with capability == (8, 0) but your GPU has capability (12, 0) (too new)

`fa2F@0.0.0` is not supported because:

operator wasn't built - see `python -m xformers.info` for more info

`cutlassF-pt` is not supported because:

requires device with capability <= (9, 0) but your GPU has capability (12, 0) (too new) 

 

 

3-1(해결방법). RTX 5090의 경우 xformers 사용하는 것이 안좋기 떄문에 ComfyUI\custom_nodes\ComfyUI-TRELLIS2-Pozzetti\trellis2 이 폴더 의 __init__.py 를 다음과 같이 수정한다. 이는 Trellis2내부에서 xformers를 사용하지 않겠다는 얘기. 위의 오류가 해결된다.

 

import sys
import torch
sys.modules["xformers"] = None

from . import models
from . import modules
from . import pipelines
from . import renderers
from . import representations
from . import utils