Yang2001 commited on
Commit
5ad00ce
·
verified ·
1 Parent(s): 2bea032

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
  }