Upload Pixal3D-T model weights and pipeline config
Browse files
ckpts/slat_flow_img2shape_dit_1_3B_1024_bf16.json
CHANGED
|
@@ -15,6 +15,7 @@
|
|
| 15 |
"qk_rms_norm": true,
|
| 16 |
"qk_rms_norm_cross": true,
|
| 17 |
"image_attn_mode": "proj",
|
| 18 |
-
"proj_in_channels": 2048
|
|
|
|
| 19 |
}
|
| 20 |
}
|
|
|
|
| 15 |
"qk_rms_norm": true,
|
| 16 |
"qk_rms_norm_cross": true,
|
| 17 |
"image_attn_mode": "proj",
|
| 18 |
+
"proj_in_channels": 2048,
|
| 19 |
+
"dtype": "bfloat16"
|
| 20 |
}
|
| 21 |
}
|
ckpts/slat_flow_img2shape_dit_1_3B_512_bf16.json
CHANGED
|
@@ -15,6 +15,7 @@
|
|
| 15 |
"qk_rms_norm": true,
|
| 16 |
"qk_rms_norm_cross": true,
|
| 17 |
"image_attn_mode": "proj",
|
| 18 |
-
"proj_in_channels": 2048
|
|
|
|
| 19 |
}
|
| 20 |
}
|
|
|
|
| 15 |
"qk_rms_norm": true,
|
| 16 |
"qk_rms_norm_cross": true,
|
| 17 |
"image_attn_mode": "proj",
|
| 18 |
+
"proj_in_channels": 2048,
|
| 19 |
+
"dtype": "bfloat16"
|
| 20 |
}
|
| 21 |
}
|
ckpts/slat_flow_imgshape2tex_dit_1_3B_1024_bf16.json
CHANGED
|
@@ -15,6 +15,7 @@
|
|
| 15 |
"qk_rms_norm": true,
|
| 16 |
"qk_rms_norm_cross": true,
|
| 17 |
"image_attn_mode": "proj",
|
| 18 |
-
"proj_in_channels": 2048
|
|
|
|
| 19 |
}
|
| 20 |
}
|
|
|
|
| 15 |
"qk_rms_norm": true,
|
| 16 |
"qk_rms_norm_cross": true,
|
| 17 |
"image_attn_mode": "proj",
|
| 18 |
+
"proj_in_channels": 2048,
|
| 19 |
+
"dtype": "bfloat16"
|
| 20 |
}
|
| 21 |
}
|
ckpts/ss_flow_img_dit_1_3B_64_bf16.json
CHANGED
|
@@ -14,6 +14,7 @@
|
|
| 14 |
"initialization": "scaled",
|
| 15 |
"qk_rms_norm": true,
|
| 16 |
"qk_rms_norm_cross": true,
|
| 17 |
-
"image_attn_mode": "proj"
|
|
|
|
| 18 |
}
|
| 19 |
}
|
|
|
|
| 14 |
"initialization": "scaled",
|
| 15 |
"qk_rms_norm": true,
|
| 16 |
"qk_rms_norm_cross": true,
|
| 17 |
+
"image_attn_mode": "proj",
|
| 18 |
+
"dtype": "bfloat16"
|
| 19 |
}
|
| 20 |
}
|