Upload folder using huggingface_hub

ft/matvae2/checkpoints/checkpoint_852/random_states_0.pkl

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

ft/matvae2/checkpoints/checkpoint_853/custom_checkpoint_0.pkl

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

ft/matvae2/checkpoints/checkpoint_853/custom_checkpoint_1.pkl

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

ft/matvae2/checkpoints/checkpoint_853/custom_checkpoint_2.pkl

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

ft/matvae2/checkpoints/checkpoint_853/optimizer.bin

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

ft/matvae2/checkpoints/checkpoint_853/random_states_0.pkl

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

ft/matvae2/final_model/model.safetensors

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

inference.json → vomp_1_0/inference.json

@@ -1,8 +1,8 @@
 {
-    "geometry_checkpoint_dir": "weights/geometry_transformer.pt",
-    "matvae_checkpoint_dir": "weights/matvae.safetensors",
-    "normalization_params_path": "weights/normalization_params.json",
-    "matvae_checkpoint": "weights/matvae.safetensors",
+    "geometry_checkpoint_dir": "weights/vomp_1_0/geometry_transformer.pt",
+    "matvae_checkpoint_dir": "weights/vomp_1_0/matvae.safetensors",
+    "normalization_params_path": "weights/vomp_1_0/normalization_params.json",
+    "matvae_checkpoint": "weights/vomp_1_0/matvae.safetensors",
     "trellis_weights_path": "weights/TRELLIS-image-large",
     "models": {
         "geometry_encoder": {
@@ -83,4 +83,4 @@
             "loss_type": "l1"
         }
     }
-}
+}

ft/matvae2/checkpoints/checkpoint_852/custom_checkpoint_0.pkl → vomp_1_1/ft/geometry_transformer/ckpts/geometry_encoder_ema0.9999_step0164000.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f622e3c5a07f4ce714d0de5722b367082314db533edffc2f3baa39777e4305aa
-size 1240
+oid sha256:533f6f1aff7da7bc101741c1bf33639b9a85aaa338fa5645ab1cc637f7f9f470
+size 343249527

ft/matvae2/checkpoints/checkpoint_852/custom_checkpoint_1.pkl → vomp_1_1/ft/geometry_transformer/ckpts/geometry_encoder_step0164000.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:625bea504e46688a1ae7c0ceb4065b53420169b636d9bc67991a5f1930e00475
-size 1240
+oid sha256:4f55e3ec674582050a9733c1fb03603ccf2f569b2309a8b80102955bf560f984
+size 343249477

ft/matvae2/checkpoints/checkpoint_852/optimizer.bin → vomp_1_1/ft/geometry_transformer/ckpts/misc_step0164000.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f209ae0cebcff7282c24ff3a9f58c6e334e20742de081c0e742c0e29ce8d388c
-size 3283962
+oid sha256:0b371871e84fd52d32e30e31a74695aee58271a6c9d4a938b86158d52209408b
+size 686465363

{ft/matvae2/checkpoints/checkpoint_853 → vomp_1_1/ft/matvae/final_model}/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b091b1760fd2fdda87c9212fbd662d72743c93a7f21e2e134cdb5cd25604817d
-size 1620768
+oid sha256:dc5746747035f60769380a74b09c5b19c8ed2a2756fe1cf4204a4ff6c1f712fe
+size 1620496

vomp_1_1/ft/matvae/normalization_params.json

@@ -0,0 +1,20 @@
+{
+  "E_min": 20.0,
+  "E_max": 1199999942656.0,
+  "nu_min": -0.19230769574642181,
+  "nu_max": 0.5,
+  "rho_min": 1.2000000476837158,
+  "rho_max": 22087.251953125,
+  "normalization_type": "log_minmax",
+  "dataset_size": 90959,
+  "train_size": 90959,
+  "val_size": 1937,
+  "unique_dataset_size": 47218,
+  "train_unique_size": 47218,
+  "heldout_source_rows": 0,
+  "validation_source_rows": 1937,
+  "validation_source_unique_size": 1937,
+  "validation_overlap_removed": 0,
+  "validation_unique_size": 1937,
+  "protected_train_unique_size": 2573
+}

vomp_1_1/geometry_transformer.pt

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

vomp_1_1/inference.json

@@ -0,0 +1,52 @@
+{
+    "geometry_checkpoint_dir": "weights/vomp_1_1/geometry_transformer.pt",
+    "matvae_checkpoint_dir": "weights/vomp_1_1/matvae.safetensors",
+    "normalization_params_path": "weights/vomp_1_1/normalization_params.json",
+    "matvae_checkpoint": "weights/vomp_1_1/matvae.safetensors",
+    "trellis_weights_path": "weights/TRELLIS-image-large",
+    "models": {
+        "geometry_encoder": {
+            "name": "geometry_encoder",
+            "args": {
+                "resolution": 64,
+                "in_channels": 1024,
+                "model_channels": 768,
+                "latent_channels": 2,
+                "num_blocks": 12,
+                "num_heads": 12,
+                "mlp_ratio": 4,
+                "attn_mode": "swin",
+                "window_size": 8,
+                "use_fp16": true
+            }
+        },
+        "matvae": {
+            "name": "StandardVAE",
+            "args": {
+                "width": 256,
+                "depth": 3,
+                "z_dim": 2,
+                "p_drop": 0.05,
+                "use_learned_variances": false,
+                "use_additional_losses": true,
+                "use_flow": false
+            }
+        }
+    },
+    "dataset": {
+        "name": "SparseVoxelMaterials",
+        "normalization_type": "log_minmax",
+        "args": {
+            "roots": "datasets/simready",
+            "image_size": 512,
+            "model": "dinov2_vitl14_reg",
+            "resolution": 64,
+            "min_aesthetic_score": 0.0,
+            "max_num_voxels": 32768,
+            "compute_material_stats": false,
+            "archive_format": "mapped",
+            "random_grid_rotation": false,
+            "segment_balanced_sampling": true
+        }
+    }
+}

ft/matvae2/checkpoints/checkpoint_851/custom_checkpoint_2.pkl → vomp_1_1/matvae.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b4203470b746c23fc434fceefb67ef86c8aa4896dabb5696c80e5146bc479acc
-size 912
+oid sha256:dc5746747035f60769380a74b09c5b19c8ed2a2756fe1cf4204a4ff6c1f712fe
+size 1620496

vomp_1_1/normalization_params.json

@@ -0,0 +1,20 @@
+{
+  "E_min": 20.0,
+  "E_max": 1199999942656.0,
+  "nu_min": -0.19230769574642181,
+  "nu_max": 0.5,
+  "rho_min": 1.2000000476837158,
+  "rho_max": 22087.251953125,
+  "normalization_type": "log_minmax",
+  "dataset_size": 90959,
+  "train_size": 90959,
+  "val_size": 1937,
+  "unique_dataset_size": 47218,
+  "train_unique_size": 47218,
+  "heldout_source_rows": 0,
+  "validation_source_rows": 1937,
+  "validation_source_unique_size": 1937,
+  "validation_overlap_removed": 0,
+  "validation_unique_size": 1937,
+  "protected_train_unique_size": 2573
+}