SymTRELLIS / README.md
quantaji's picture
dd
7181d40
|
Raw
History Blame Contribute Delete
5.36 kB
---
license: cc-by-nc-4.0
pipeline_tag: image-to-3d
---
# SymTRELLIS
This repository hosts model artifacts for the paper **SymTRELLIS: Symmetry-Enforced Voxel Latents for 3D Generation**.
([arxiv](https://arxiv.org/abs/2606.04108)|[code](https://github.com/symtrellis/SymTRELLIS)|[project](https://symtrellis.github.io)|[Training Data](https://huggingface.co/datasets/symtrellis/SymTRELLIS-Latent-Transform-Pairs)|[Demo](https://huggingface.co/spaces/quantaji/SymTRELLIS))
## Model Zoo
<table>
<thead>
<tr>
<th>Base Model</th>
<th>Mapper Stage</th>
<th>Backend</th>
<th>Train</th>
<th>Links</th>
</tr>
</thead>
<tbody>
<tr>
<td rowspan="8">TRELLIS.2</td>
<td rowspan="4">Sparse Structure</td>
<td rowspan="2">Neighbor Graph</td>
<td>Pretrain</td>
<td>
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/neighbor_graph/pretrain/config.json">model config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/neighbor_graph/pretrain/train.sh">train config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/neighbor_graph/pretrain/model.safetensors">weights</a>
</td>
</tr>
<tr>
<td>Finetune</td>
<td>
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/neighbor_graph/finetune/config.json">model config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/neighbor_graph/finetune/train.sh">train config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/neighbor_graph/finetune/model.safetensors">weights</a>
</td>
</tr>
<tr>
<td rowspan="2">Swin3D</td>
<td>Pretrain</td>
<td>
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/swin3d/pretrain/config.json">model config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/swin3d/pretrain/train.sh">train config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/swin3d/pretrain/model.safetensors">weights</a>
</td>
</tr>
<tr>
<td>Finetune</td>
<td>
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/swin3d/finetune/config.json">model config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/swin3d/finetune/train.sh">train config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/sparse_structure/swin3d/finetune/model.safetensors">weights</a>
</td>
</tr>
<tr>
<td rowspan="4">Shape</td>
<td rowspan="2">Neighbor Graph</td>
<td>Pretrain</td>
<td>
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/shape/neighbor_graph/pretrain/config.json">model config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/shape/neighbor_graph/pretrain/train.sh">train config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/shape/neighbor_graph/pretrain/model.safetensors">weights</a>
</td>
</tr>
<tr>
<td>Finetune</td>
<td>
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/shape/neighbor_graph/finetune/config.json">model config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/shape/neighbor_graph/finetune/train.sh">train config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/shape/neighbor_graph/finetune/model.safetensors">weights</a>
</td>
</tr>
<tr>
<td rowspan="2">Swin3D</td>
<td>Pretrain</td>
<td>
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/shape/swin3d/pretrain/config.json">model config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/shape/swin3d/pretrain/train.sh">train config</a> /
<a href="https://huggingface.co/quantaji/SymTRELLIS/blob/main/trellis2/shape/swin3d/pretrain/model.safetensors">weights</a>
</td>
</tr>
<tr>
<td>Finetune</td>
<td>-</td>
</tr>
<tr>
<td rowspan="5">TRELLIS</td>
<td>Sparse Structure</td>
<td colspan="3">VAE and Mapper shared with TRELLIS.2</td>
</tr>
<tr>
<td rowspan="4">Shape</td>
<td rowspan="2">Neighbor Graph</td>
<td>Pretrain</td>
<td>-</td>
</tr>
<tr>
<td>Finetune</td>
<td>-</td>
</tr>
<tr>
<td rowspan="2">Swin3D</td>
<td>Pretrain</td>
<td>-</td>
</tr>
<tr>
<td>Finetune</td>
<td>-</td>
</tr>
<tr>
<td rowspan="4">SAM 3D Objects</td>
<td rowspan="4">Shape</td>
<td rowspan="2">Neighbor Graph</td>
<td>Pretrain</td>
<td>-</td>
</tr>
<tr>
<td>Finetune</td>
<td>-</td>
</tr>
<tr>
<td rowspan="2">Swin3D</td>
<td>Pretrain</td>
<td>-</td>
</tr>
<tr>
<td>Finetune</td>
<td>-</td>
</tr>
</tbody>
</table>