Trellis2 가지고 놀아보기
https://jeffreyxiang.github.io/blogs
Trellis2를 만든 존경할 만한 사람이다.
틈틈히 이 사람의 블로그 게시물을 탐독하리라
https://github.com/microsoft/TRELLIS.2
[WSL2]
git clone -b main https://github.com/microsoft/TRELLIS.2.git --recursive
cd TRELLIS.2
. ./setup.sh --new-env --basic --flash-attn --nvdiffrast --nvdiffrec --cumesh --o-voxel --flexgemm --new-env
y
#pillow 관련 라이브러리가 꼬였을 때에는 아래 두 커맨드로 다시 설치한다.
pip uninstall pillow -y
pip install pillow
#
pip install psutil ninja
pip install flash-attn==2.7.4.post1 --no-build-isolation
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
python app.py
그리고
처럼 내부적으로 사용되는 모델에 대한 접근 권한을 획득해두시라
flex_gemm은 JeffreyXiang의 블로그에 있던 건데 역시 대단한거 같다.
실행하면 이런식으로 gradio webui가 뜨게 된다.
근데 cuda OOM 에러가 났다…
https://github.com/fox1245/TRELLIS.2
위의 링크는 나의 후진 GPU에서도 돌아갈 수 있게끔 수정한 포크 버전이다.
다음과 같이 성공했다.
타인이형 얼짱각도