Upload folder using huggingface_hub

conditioner/conditioner.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:730518ae5f2b6af51db12137ed0dc0ac791ee04d927cf8e4f2043db9c8218251
+size 1755920004

conditioner/config.json

@@ -0,0 +1,65 @@
+{
+    "target": "partgen.models.conditioner.condioner_release.Conditioner",
+    "params": {
+        "use_geo": true,
+        "use_obj": true,
+        "use_seg_feat": true,
+        "geo_cfg": {
+            "target": "partgen.models.conditioner.part_encoders.PartEncoder",
+            "output_dim": 1024,
+            "params": {
+                "use_local": true,
+                "local_feat_type": "latents_shape",
+                "num_tokens_cond": 4096,
+                "local_geo_cfg": {
+                    "target": "partgen.models.autoencoders.VolumeDecoderShapeVAE",
+                    "params": {
+                        "num_latents": 4096,
+                        "embed_dim": 64,
+                        "num_freqs": 8,
+                        "include_pi": false,
+                        "heads": 16,
+                        "width": 1024,
+                        "num_encoder_layers": 8,
+                        "num_decoder_layers": 16,
+                        "qkv_bias": false,
+                        "qk_norm": true,
+                        "scale_factor": null,
+                        "geo_decoder_mlp_expand_ratio": 4,
+                        "geo_decoder_downsample_ratio": 1,
+                        "geo_decoder_ln_post": true,
+                        "point_feats": 4,
+                        "pc_size": 81920,
+                        "pc_sharpedge_size": 0
+                    }
+                }
+            }
+        },
+        "obj_encoder_cfg": {
+            "target": "partgen.models.autoencoders.VolumeDecoderShapeVAE",
+            "output_dim": 1024,
+            "params": {
+                "num_latents": 4096,
+                "embed_dim": 64,
+                "num_freqs": 8,
+                "include_pi": false,
+                "heads": 16,
+                "width": 1024,
+                "num_encoder_layers": 8,
+                "num_decoder_layers": 16,
+                "qkv_bias": false,
+                "qk_norm": true,
+                "scale_factor": 1.0039506158752403,
+                "geo_decoder_mlp_expand_ratio": 4,
+                "geo_decoder_downsample_ratio": 1,
+                "geo_decoder_ln_post": true,
+                "point_feats": 4,
+                "pc_size": 81920,
+                "pc_sharpedge_size": 0
+            }
+        },
+        "seg_feat_cfg": {
+            "target": "partgen.models.conditioner.sonata_extractor.SonataFeatureExtractor"
+        }
+    }
+}

model/config.json

@@ -0,0 +1,30 @@
+{
+    "target": "partgen.models.partformer_dit.PartFormerDITPlain",
+    "params": {
+        "use_self_attention": true,
+        "use_cross_attention": true,
+        "use_cross_attention_2": true,
+        "use_bbox_cond": false,
+        "num_freqs": 8,
+        "use_part_embed": true,
+        "valid_num": 50,
+        "input_size": "*num_latents",
+        "in_channels": 64,
+        "hidden_size": 2048,
+        "encoder_hidden_dim": "*cross1_output_dim",
+        "encoder_hidden2_dim": "*cross2_output_dim",
+        "depth": 21,
+        "num_heads": 16,
+        "qk_norm": true,
+        "qkv_bias": false,
+        "qk_norm_type": "rms",
+        "with_decoupled_ca": false,
+        "decoupled_ca_dim": "*num_tokens_cond",
+        "decoupled_ca_weight": 1.0,
+        "use_attention_pooling": false,
+        "use_pos_emb": false,
+        "num_moe_layers": 6,
+        "num_experts": 8,
+        "moe_top_k": 2
+    }
+}

model/model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4ca47998e1357432302d7de0ec85f7e4076b8e64037935f033b881e2fd9973e0
+size 6630571584

p3sam.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5c6555ea891939a746512fe2a0b22f861d8e0910aa24b7981c63ea336fa10090
+size 1353382727

p3sam/config.json

@@ -0,0 +1,6 @@
+{
+    "target": "partgen.bbox_estimator.auto_mask_api.AutoMask",
+    "params": {
+        "ckpt_path": "checkpoints/p3sam.ckpt"
+    }
+}

p3sam/p3sam.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:eb76550cfbe06f154c6e9b17167ccfc28222bb4a216ec7b12ac2bf7d762de38c
+size 450968044

scheduler/config.json

@@ -0,0 +1,6 @@
+{
+    "target": "partgen.models.diffusion.schedulers.FlowMatchEulerDiscreteScheduler",
+    "params": {
+        "num_train_timesteps": 1000
+    }
+}

shapevae/config.json

@@ -0,0 +1,22 @@
+{
+    "target": "partgen.models.autoencoders.VolumeDecoderShapeVAE",
+    "params": {
+        "num_latents": 1024,
+        "embed_dim": 64,
+        "num_freqs": 8,
+        "include_pi": false,
+        "heads": 16,
+        "width": 1024,
+        "num_encoder_layers": 8,
+        "num_decoder_layers": 16,
+        "qkv_bias": false,
+        "qk_norm": true,
+        "scale_factor": 1.0039506158752403,
+        "geo_decoder_mlp_expand_ratio": 4,
+        "geo_decoder_downsample_ratio": 1,
+        "geo_decoder_ln_post": true,
+        "point_feats": 4,
+        "pc_size": 81920,
+        "pc_sharpedge_size": 0
+    }
+}

shapevae/shapevae.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3d46179a603c529c42ce8c1eac6d56667700b1f16626e4bc5bc319c99a4f3b88
+size 655539474