Upload 56 files
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.json +31 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.safetensors +3 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.json +17 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.safetensors +3 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.json +18 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.safetensors +3 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/slat_flow_img_dit_L_64l8p2_fp16.json +19 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/slat_flow_img_dit_L_64l8p2_fp16.safetensors +3 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16.json +12 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16.safetensors +3 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/ss_flow_img_dit_L_16l8_fp16.json +17 -0
- JeffreyXiang/TRELLIS-image-large/ckpts/ss_flow_img_dit_L_16l8_fp16.safetensors +3 -0
- JeffreyXiang/TRELLIS-image-large/pipeline.json +60 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/.gitignore +1 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/.gitattributes.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/README.md.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.json.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.safetensors.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.json.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.safetensors.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.json.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.safetensors.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_enc_swin8_B_64l8_fp16.json.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_enc_swin8_B_64l8_fp16.safetensors.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_flow_img_dit_L_64l8p2_fp16.json.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_flow_img_dit_L_64l8p2_fp16.safetensors.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_dec_conv3d_16l8_fp16.json.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_dec_conv3d_16l8_fp16.safetensors.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_enc_conv3d_16l8_fp16.json.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_enc_conv3d_16l8_fp16.safetensors.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_flow_img_dit_L_16l8_fp16.json.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_flow_img_dit_L_16l8_fp16.safetensors.metadata +3 -0
- checkpoints/TRELLIS-image-large/.cache/huggingface/download/pipeline.json.metadata +3 -0
- checkpoints/TRELLIS-image-large/.gitattributes +35 -0
- checkpoints/TRELLIS-image-large/README.md +16 -0
- checkpoints/TRELLIS-image-large/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.json +31 -0
- checkpoints/TRELLIS-image-large/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.safetensors +3 -0
- checkpoints/TRELLIS-image-large/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.json +17 -0
- checkpoints/TRELLIS-image-large/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.safetensors +3 -0
- checkpoints/TRELLIS-image-large/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.json +18 -0
- checkpoints/TRELLIS-image-large/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.safetensors +3 -0
- checkpoints/TRELLIS-image-large/ckpts/slat_enc_swin8_B_64l8_fp16.json +15 -0
- checkpoints/TRELLIS-image-large/ckpts/slat_enc_swin8_B_64l8_fp16.safetensors +3 -0
- checkpoints/TRELLIS-image-large/ckpts/slat_flow_img_dit_L_64l8p2_fp16.json +19 -0
- checkpoints/TRELLIS-image-large/ckpts/slat_flow_img_dit_L_64l8p2_fp16.safetensors +3 -0
- checkpoints/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16.json +12 -0
- checkpoints/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16.safetensors +3 -0
- checkpoints/TRELLIS-image-large/ckpts/ss_enc_conv3d_16l8_fp16.json +12 -0
- checkpoints/TRELLIS-image-large/ckpts/ss_enc_conv3d_16l8_fp16.safetensors +3 -0
- checkpoints/TRELLIS-image-large/ckpts/ss_flow_img_dit_L_16l8_fp16.json +17 -0
JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.json
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SLatGaussianDecoder",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 64,
|
| 5 |
+
"model_channels": 768,
|
| 6 |
+
"latent_channels": 8,
|
| 7 |
+
"num_blocks": 12,
|
| 8 |
+
"num_heads": 12,
|
| 9 |
+
"mlp_ratio": 4,
|
| 10 |
+
"attn_mode": "swin",
|
| 11 |
+
"window_size": 8,
|
| 12 |
+
"use_fp16": true,
|
| 13 |
+
"representation_config": {
|
| 14 |
+
"lr": {
|
| 15 |
+
"_xyz": 1.0,
|
| 16 |
+
"_features_dc": 1.0,
|
| 17 |
+
"_opacity": 1.0,
|
| 18 |
+
"_scaling": 1.0,
|
| 19 |
+
"_rotation": 0.1
|
| 20 |
+
},
|
| 21 |
+
"perturb_offset": true,
|
| 22 |
+
"voxel_size": 1.5,
|
| 23 |
+
"num_gaussians": 32,
|
| 24 |
+
"2d_filter_kernel_size": 0.1,
|
| 25 |
+
"3d_filter_kernel_size": 9e-4,
|
| 26 |
+
"scaling_bias": 4e-3,
|
| 27 |
+
"opacity_bias": 0.1,
|
| 28 |
+
"scaling_activation": "softplus"
|
| 29 |
+
}
|
| 30 |
+
}
|
| 31 |
+
}
|
JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:38c84bcef5ce0af1f48b1b5558dabc7575a13346043c41a7e0610f1fa619a161
|
| 3 |
+
size 171450952
|
JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SLatMeshDecoder",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 64,
|
| 5 |
+
"model_channels": 768,
|
| 6 |
+
"latent_channels": 8,
|
| 7 |
+
"num_blocks": 12,
|
| 8 |
+
"num_heads": 12,
|
| 9 |
+
"mlp_ratio": 4,
|
| 10 |
+
"attn_mode": "swin",
|
| 11 |
+
"window_size": 8,
|
| 12 |
+
"use_fp16": true,
|
| 13 |
+
"representation_config": {
|
| 14 |
+
"use_color": true
|
| 15 |
+
}
|
| 16 |
+
}
|
| 17 |
+
}
|
JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3e87aba94b5786407eb06d0502c1ed0885a0027a3f2b8537bfe15b0a92c01859
|
| 3 |
+
size 181903412
|
JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SLatRadianceFieldDecoder",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 64,
|
| 5 |
+
"model_channels": 768,
|
| 6 |
+
"latent_channels": 8,
|
| 7 |
+
"num_blocks": 12,
|
| 8 |
+
"num_heads": 12,
|
| 9 |
+
"mlp_ratio": 4,
|
| 10 |
+
"attn_mode": "swin",
|
| 11 |
+
"window_size": 8,
|
| 12 |
+
"use_fp16": true,
|
| 13 |
+
"representation_config": {
|
| 14 |
+
"rank": 16,
|
| 15 |
+
"dim": 8
|
| 16 |
+
}
|
| 17 |
+
}
|
| 18 |
+
}
|
JeffreyXiang/TRELLIS-image-large/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:737da6578d01948016b7c39786113af0d64a46f7922f6b8b5e698b84643be514
|
| 3 |
+
size 171450488
|
JeffreyXiang/TRELLIS-image-large/ckpts/slat_flow_img_dit_L_64l8p2_fp16.json
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SLatFlowModel",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 64,
|
| 5 |
+
"in_channels": 8,
|
| 6 |
+
"out_channels": 8,
|
| 7 |
+
"model_channels": 1024,
|
| 8 |
+
"cond_channels": 1024,
|
| 9 |
+
"num_blocks": 24,
|
| 10 |
+
"num_heads": 16,
|
| 11 |
+
"mlp_ratio": 4,
|
| 12 |
+
"patch_size": 2,
|
| 13 |
+
"num_io_res_blocks": 2,
|
| 14 |
+
"io_block_channels": [128],
|
| 15 |
+
"pe_mode": "ape",
|
| 16 |
+
"qk_rms_norm": true,
|
| 17 |
+
"use_fp16": true
|
| 18 |
+
}
|
| 19 |
+
}
|
JeffreyXiang/TRELLIS-image-large/ckpts/slat_flow_img_dit_L_64l8p2_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:693fb2a58ad497bd222007301eeec49d14d60f8c12d2f2f00c221fa747b4c66c
|
| 3 |
+
size 1203755136
|
JeffreyXiang/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16.json
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
{
|
| 3 |
+
"name": "SparseStructureDecoder",
|
| 4 |
+
"args": {
|
| 5 |
+
"out_channels": 1,
|
| 6 |
+
"latent_channels": 8,
|
| 7 |
+
"num_res_blocks": 2,
|
| 8 |
+
"num_res_blocks_middle": 2,
|
| 9 |
+
"channels": [512, 128, 32],
|
| 10 |
+
"use_fp16": true
|
| 11 |
+
}
|
| 12 |
+
}
|
JeffreyXiang/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1c76d4a40519aa2d711cc263a8404105231ac26db31d946bed48b84fee79009a
|
| 3 |
+
size 147591972
|
JeffreyXiang/TRELLIS-image-large/ckpts/ss_flow_img_dit_L_16l8_fp16.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SparseStructureFlowModel",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 16,
|
| 5 |
+
"in_channels": 8,
|
| 6 |
+
"out_channels": 8,
|
| 7 |
+
"model_channels": 1024,
|
| 8 |
+
"cond_channels": 1024,
|
| 9 |
+
"num_blocks": 24,
|
| 10 |
+
"num_heads": 16,
|
| 11 |
+
"mlp_ratio": 4,
|
| 12 |
+
"patch_size": 1,
|
| 13 |
+
"pe_mode": "ape",
|
| 14 |
+
"qk_rms_norm": true,
|
| 15 |
+
"use_fp16": true
|
| 16 |
+
}
|
| 17 |
+
}
|
JeffreyXiang/TRELLIS-image-large/ckpts/ss_flow_img_dit_L_16l8_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:96dc6bfd4136fd950af564dd16b4ae533c9ba6af8f26c670646b2a9f2789b1db
|
| 3 |
+
size 1130770840
|
JeffreyXiang/TRELLIS-image-large/pipeline.json
ADDED
|
@@ -0,0 +1,60 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "TrellisImageTo3DPipeline",
|
| 3 |
+
"args": {
|
| 4 |
+
"models": {
|
| 5 |
+
"sparse_structure_decoder": "ckpts/ss_dec_conv3d_16l8_fp16",
|
| 6 |
+
"sparse_structure_flow_model": "ckpts/ss_flow_img_dit_L_16l8_fp16",
|
| 7 |
+
"slat_decoder_gs": "ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16",
|
| 8 |
+
"slat_decoder_rf": "ckpts/slat_dec_rf_swin8_B_64l8r16_fp16",
|
| 9 |
+
"slat_decoder_mesh": "ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16",
|
| 10 |
+
"slat_flow_model": "ckpts/slat_flow_img_dit_L_64l8p2_fp16"
|
| 11 |
+
},
|
| 12 |
+
"sparse_structure_sampler": {
|
| 13 |
+
"name": "FlowEulerGuidanceIntervalSampler",
|
| 14 |
+
"args": {
|
| 15 |
+
"sigma_min": 1e-5
|
| 16 |
+
},
|
| 17 |
+
"params": {
|
| 18 |
+
"steps": 25,
|
| 19 |
+
"cfg_strength": 5.0,
|
| 20 |
+
"cfg_interval": [0.5, 1.0],
|
| 21 |
+
"rescale_t": 3.0
|
| 22 |
+
}
|
| 23 |
+
},
|
| 24 |
+
"slat_sampler": {
|
| 25 |
+
"name": "FlowEulerGuidanceIntervalSampler",
|
| 26 |
+
"args": {
|
| 27 |
+
"sigma_min": 1e-5
|
| 28 |
+
},
|
| 29 |
+
"params": {
|
| 30 |
+
"steps": 25,
|
| 31 |
+
"cfg_strength": 5.0,
|
| 32 |
+
"cfg_interval": [0.5, 1.0],
|
| 33 |
+
"rescale_t": 3.0
|
| 34 |
+
}
|
| 35 |
+
},
|
| 36 |
+
"slat_normalization": {
|
| 37 |
+
"mean": [
|
| 38 |
+
-2.1687545776367188,
|
| 39 |
+
-0.004347046371549368,
|
| 40 |
+
-0.13352349400520325,
|
| 41 |
+
-0.08418072760105133,
|
| 42 |
+
-0.5271206498146057,
|
| 43 |
+
0.7238689064979553,
|
| 44 |
+
-1.1414450407028198,
|
| 45 |
+
1.2039363384246826
|
| 46 |
+
],
|
| 47 |
+
"std": [
|
| 48 |
+
2.377650737762451,
|
| 49 |
+
2.386378288269043,
|
| 50 |
+
2.124418020248413,
|
| 51 |
+
2.1748552322387695,
|
| 52 |
+
2.663944721221924,
|
| 53 |
+
2.371192216873169,
|
| 54 |
+
2.6217446327209473,
|
| 55 |
+
2.684523105621338
|
| 56 |
+
]
|
| 57 |
+
},
|
| 58 |
+
"image_cond_model": "dinov2_vitl14_reg"
|
| 59 |
+
}
|
| 60 |
+
}
|
checkpoints/TRELLIS-image-large/.cache/huggingface/.gitignore
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
*
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/.gitattributes.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
a6344aac8c09253b3b630fb776ae94478aa0275b
|
| 3 |
+
1740450455.5786986
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/README.md.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
e2979a10b76f2cd9f2ea3f12d42f3c4e4d0b00bc
|
| 3 |
+
1740450456.0814295
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.json.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
051ade1e3b374738bf7a007516275f72e458f2a7
|
| 3 |
+
1740450455.5533767
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.safetensors.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
38c84bcef5ce0af1f48b1b5558dabc7575a13346043c41a7e0610f1fa619a161
|
| 3 |
+
1740450477.4681325
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.json.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
28802825e9ba04bcc2cfdffb491dc9dbb72c8a38
|
| 3 |
+
1740450455.5786986
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.safetensors.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
3e87aba94b5786407eb06d0502c1ed0885a0027a3f2b8537bfe15b0a92c01859
|
| 3 |
+
1740450477.2603555
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.json.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
10676bd6322cb1519c9cb603ea77cfa2057ae45e
|
| 3 |
+
1740450455.8506227
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.safetensors.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
737da6578d01948016b7c39786113af0d64a46f7922f6b8b5e698b84643be514
|
| 3 |
+
1740450475.0393395
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_enc_swin8_B_64l8_fp16.json.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
19b9b8c99c3b56eb9f07066d2a4bbad578b10291
|
| 3 |
+
1740450456.0915627
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_enc_swin8_B_64l8_fp16.safetensors.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
21dceac6bee917ab6458ff52c9757ba89a779d03031c7bd17f9e7f0103bfd436
|
| 3 |
+
1740450478.7897506
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_flow_img_dit_L_64l8p2_fp16.json.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
22f500ab89867eb147f04d313cffac98e54ca163
|
| 3 |
+
1740450456.1939182
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/slat_flow_img_dit_L_64l8p2_fp16.safetensors.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
693fb2a58ad497bd222007301eeec49d14d60f8c12d2f2f00c221fa747b4c66c
|
| 3 |
+
1740450531.3041167
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_dec_conv3d_16l8_fp16.json.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
9f3affaf13ab29fe48105229da9fab72ea8de716
|
| 3 |
+
1740450456.6326063
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_dec_conv3d_16l8_fp16.safetensors.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
1c76d4a40519aa2d711cc263a8404105231ac26db31d946bed48b84fee79009a
|
| 3 |
+
1740450473.947149
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_enc_conv3d_16l8_fp16.json.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
af493f4615283446a71d281eb4fb5d87cc32f159
|
| 3 |
+
1740450456.7103057
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_enc_conv3d_16l8_fp16.safetensors.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
107874eeaa0feb82f51b19db5da7db534fb7e7f19e5a122b9ff1bc2e258bfc6d
|
| 3 |
+
1740450474.139573
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_flow_img_dit_L_16l8_fp16.json.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
d228f99cd828a54695ebd1f0fd1ad1b1f3179f42
|
| 3 |
+
1740450457.2410107
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/ckpts/ss_flow_img_dit_L_16l8_fp16.safetensors.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
96dc6bfd4136fd950af564dd16b4ae533c9ba6af8f26c670646b2a9f2789b1db
|
| 3 |
+
1740450528.5150473
|
checkpoints/TRELLIS-image-large/.cache/huggingface/download/pipeline.json.metadata
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
25e0d31ffbebe4b5a97464dd851910efc3002d96
|
| 2 |
+
4141e1025889c44ec0dc9239157c4226d8387da0
|
| 3 |
+
1740450474.4769437
|
checkpoints/TRELLIS-image-large/.gitattributes
ADDED
|
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
*.7z filter=lfs diff=lfs merge=lfs -text
|
| 2 |
+
*.arrow filter=lfs diff=lfs merge=lfs -text
|
| 3 |
+
*.bin filter=lfs diff=lfs merge=lfs -text
|
| 4 |
+
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
| 5 |
+
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
| 6 |
+
*.ftz filter=lfs diff=lfs merge=lfs -text
|
| 7 |
+
*.gz filter=lfs diff=lfs merge=lfs -text
|
| 8 |
+
*.h5 filter=lfs diff=lfs merge=lfs -text
|
| 9 |
+
*.joblib filter=lfs diff=lfs merge=lfs -text
|
| 10 |
+
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
| 11 |
+
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
| 12 |
+
*.model filter=lfs diff=lfs merge=lfs -text
|
| 13 |
+
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
| 14 |
+
*.npy filter=lfs diff=lfs merge=lfs -text
|
| 15 |
+
*.npz filter=lfs diff=lfs merge=lfs -text
|
| 16 |
+
*.onnx filter=lfs diff=lfs merge=lfs -text
|
| 17 |
+
*.ot filter=lfs diff=lfs merge=lfs -text
|
| 18 |
+
*.parquet filter=lfs diff=lfs merge=lfs -text
|
| 19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
| 20 |
+
*.pickle filter=lfs diff=lfs merge=lfs -text
|
| 21 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
| 22 |
+
*.pt filter=lfs diff=lfs merge=lfs -text
|
| 23 |
+
*.pth filter=lfs diff=lfs merge=lfs -text
|
| 24 |
+
*.rar filter=lfs diff=lfs merge=lfs -text
|
| 25 |
+
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
| 26 |
+
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
| 27 |
+
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
| 28 |
+
*.tar filter=lfs diff=lfs merge=lfs -text
|
| 29 |
+
*.tflite filter=lfs diff=lfs merge=lfs -text
|
| 30 |
+
*.tgz filter=lfs diff=lfs merge=lfs -text
|
| 31 |
+
*.wasm filter=lfs diff=lfs merge=lfs -text
|
| 32 |
+
*.xz filter=lfs diff=lfs merge=lfs -text
|
| 33 |
+
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
+
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
+
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
checkpoints/TRELLIS-image-large/README.md
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: trellis
|
| 3 |
+
pipeline_tag: image-to-3d
|
| 4 |
+
license: mit
|
| 5 |
+
language:
|
| 6 |
+
- en
|
| 7 |
+
---
|
| 8 |
+
# TRELLIS Image Large
|
| 9 |
+
|
| 10 |
+
<!-- Provide a quick summary of what the model is/does. -->
|
| 11 |
+
|
| 12 |
+
The image conditioned version of TRELLIS, a large 3D genetive model. It was introduced in the paper [Structured 3D Latents for Scalable and Versatile 3D Generation](https://huggingface.co/papers/2412.01506).
|
| 13 |
+
|
| 14 |
+
Project page: https://trellis3d.github.io/
|
| 15 |
+
|
| 16 |
+
Code: https://github.com/Microsoft/TRELLIS
|
checkpoints/TRELLIS-image-large/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.json
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SLatGaussianDecoder",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 64,
|
| 5 |
+
"model_channels": 768,
|
| 6 |
+
"latent_channels": 8,
|
| 7 |
+
"num_blocks": 12,
|
| 8 |
+
"num_heads": 12,
|
| 9 |
+
"mlp_ratio": 4,
|
| 10 |
+
"attn_mode": "swin",
|
| 11 |
+
"window_size": 8,
|
| 12 |
+
"use_fp16": true,
|
| 13 |
+
"representation_config": {
|
| 14 |
+
"lr": {
|
| 15 |
+
"_xyz": 1.0,
|
| 16 |
+
"_features_dc": 1.0,
|
| 17 |
+
"_opacity": 1.0,
|
| 18 |
+
"_scaling": 1.0,
|
| 19 |
+
"_rotation": 0.1
|
| 20 |
+
},
|
| 21 |
+
"perturb_offset": true,
|
| 22 |
+
"voxel_size": 1.5,
|
| 23 |
+
"num_gaussians": 32,
|
| 24 |
+
"2d_filter_kernel_size": 0.1,
|
| 25 |
+
"3d_filter_kernel_size": 9e-4,
|
| 26 |
+
"scaling_bias": 4e-3,
|
| 27 |
+
"opacity_bias": 0.1,
|
| 28 |
+
"scaling_activation": "softplus"
|
| 29 |
+
}
|
| 30 |
+
}
|
| 31 |
+
}
|
checkpoints/TRELLIS-image-large/ckpts/slat_dec_gs_swin8_B_64l8gs32_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:38c84bcef5ce0af1f48b1b5558dabc7575a13346043c41a7e0610f1fa619a161
|
| 3 |
+
size 171450952
|
checkpoints/TRELLIS-image-large/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SLatMeshDecoder",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 64,
|
| 5 |
+
"model_channels": 768,
|
| 6 |
+
"latent_channels": 8,
|
| 7 |
+
"num_blocks": 12,
|
| 8 |
+
"num_heads": 12,
|
| 9 |
+
"mlp_ratio": 4,
|
| 10 |
+
"attn_mode": "swin",
|
| 11 |
+
"window_size": 8,
|
| 12 |
+
"use_fp16": true,
|
| 13 |
+
"representation_config": {
|
| 14 |
+
"use_color": true
|
| 15 |
+
}
|
| 16 |
+
}
|
| 17 |
+
}
|
checkpoints/TRELLIS-image-large/ckpts/slat_dec_mesh_swin8_B_64l8m256c_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3e87aba94b5786407eb06d0502c1ed0885a0027a3f2b8537bfe15b0a92c01859
|
| 3 |
+
size 181903412
|
checkpoints/TRELLIS-image-large/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.json
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SLatRadianceFieldDecoder",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 64,
|
| 5 |
+
"model_channels": 768,
|
| 6 |
+
"latent_channels": 8,
|
| 7 |
+
"num_blocks": 12,
|
| 8 |
+
"num_heads": 12,
|
| 9 |
+
"mlp_ratio": 4,
|
| 10 |
+
"attn_mode": "swin",
|
| 11 |
+
"window_size": 8,
|
| 12 |
+
"use_fp16": true,
|
| 13 |
+
"representation_config": {
|
| 14 |
+
"rank": 16,
|
| 15 |
+
"dim": 8
|
| 16 |
+
}
|
| 17 |
+
}
|
| 18 |
+
}
|
checkpoints/TRELLIS-image-large/ckpts/slat_dec_rf_swin8_B_64l8r16_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:737da6578d01948016b7c39786113af0d64a46f7922f6b8b5e698b84643be514
|
| 3 |
+
size 171450488
|
checkpoints/TRELLIS-image-large/ckpts/slat_enc_swin8_B_64l8_fp16.json
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SLatEncoder",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 64,
|
| 5 |
+
"in_channels": 1024,
|
| 6 |
+
"model_channels": 768,
|
| 7 |
+
"latent_channels": 8,
|
| 8 |
+
"num_blocks": 12,
|
| 9 |
+
"num_heads": 12,
|
| 10 |
+
"mlp_ratio": 4,
|
| 11 |
+
"attn_mode": "swin",
|
| 12 |
+
"window_size": 8,
|
| 13 |
+
"use_fp16": true
|
| 14 |
+
}
|
| 15 |
+
}
|
checkpoints/TRELLIS-image-large/ckpts/slat_enc_swin8_B_64l8_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:21dceac6bee917ab6458ff52c9757ba89a779d03031c7bd17f9e7f0103bfd436
|
| 3 |
+
size 173242816
|
checkpoints/TRELLIS-image-large/ckpts/slat_flow_img_dit_L_64l8p2_fp16.json
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SLatFlowModel",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 64,
|
| 5 |
+
"in_channels": 8,
|
| 6 |
+
"out_channels": 8,
|
| 7 |
+
"model_channels": 1024,
|
| 8 |
+
"cond_channels": 1024,
|
| 9 |
+
"num_blocks": 24,
|
| 10 |
+
"num_heads": 16,
|
| 11 |
+
"mlp_ratio": 4,
|
| 12 |
+
"patch_size": 2,
|
| 13 |
+
"num_io_res_blocks": 2,
|
| 14 |
+
"io_block_channels": [128],
|
| 15 |
+
"pe_mode": "ape",
|
| 16 |
+
"qk_rms_norm": true,
|
| 17 |
+
"use_fp16": true
|
| 18 |
+
}
|
| 19 |
+
}
|
checkpoints/TRELLIS-image-large/ckpts/slat_flow_img_dit_L_64l8p2_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:693fb2a58ad497bd222007301eeec49d14d60f8c12d2f2f00c221fa747b4c66c
|
| 3 |
+
size 1203755136
|
checkpoints/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16.json
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
{
|
| 3 |
+
"name": "SparseStructureDecoder",
|
| 4 |
+
"args": {
|
| 5 |
+
"out_channels": 1,
|
| 6 |
+
"latent_channels": 8,
|
| 7 |
+
"num_res_blocks": 2,
|
| 8 |
+
"num_res_blocks_middle": 2,
|
| 9 |
+
"channels": [512, 128, 32],
|
| 10 |
+
"use_fp16": true
|
| 11 |
+
}
|
| 12 |
+
}
|
checkpoints/TRELLIS-image-large/ckpts/ss_dec_conv3d_16l8_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1c76d4a40519aa2d711cc263a8404105231ac26db31d946bed48b84fee79009a
|
| 3 |
+
size 147591972
|
checkpoints/TRELLIS-image-large/ckpts/ss_enc_conv3d_16l8_fp16.json
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
{
|
| 3 |
+
"name": "SparseStructureEncoder",
|
| 4 |
+
"args": {
|
| 5 |
+
"in_channels": 1,
|
| 6 |
+
"latent_channels": 8,
|
| 7 |
+
"num_res_blocks": 2,
|
| 8 |
+
"num_res_blocks_middle": 2,
|
| 9 |
+
"channels": [32, 128, 512],
|
| 10 |
+
"use_fp16": true
|
| 11 |
+
}
|
| 12 |
+
}
|
checkpoints/TRELLIS-image-large/ckpts/ss_enc_conv3d_16l8_fp16.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:107874eeaa0feb82f51b19db5da7db534fb7e7f19e5a122b9ff1bc2e258bfc6d
|
| 3 |
+
size 119068016
|
checkpoints/TRELLIS-image-large/ckpts/ss_flow_img_dit_L_16l8_fp16.json
ADDED
|
@@ -0,0 +1,17 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"name": "SparseStructureFlowModel",
|
| 3 |
+
"args": {
|
| 4 |
+
"resolution": 16,
|
| 5 |
+
"in_channels": 8,
|
| 6 |
+
"out_channels": 8,
|
| 7 |
+
"model_channels": 1024,
|
| 8 |
+
"cond_channels": 1024,
|
| 9 |
+
"num_blocks": 24,
|
| 10 |
+
"num_heads": 16,
|
| 11 |
+
"mlp_ratio": 4,
|
| 12 |
+
"patch_size": 1,
|
| 13 |
+
"pe_mode": "ape",
|
| 14 |
+
"qk_rms_norm": true,
|
| 15 |
+
"use_fp16": true
|
| 16 |
+
}
|
| 17 |
+
}
|