Upload 4 files

Add the pre-trained model of SQuadGen

ckpts/sqdiffuse.safetensors

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

ckpts/sqdiffuse.yaml

@@ -0,0 +1,76 @@
+model:
+  params:
+    depth: 28
+    n_heads: 16
+    d_head: 72
+    condition_global_keys_list: []
+    is_last_feature: 1
+    qk_norm: 1
+    is_pecode_latent: 1
+    is_pecode_latent_pe: 1
+    target_mu:
+      - -0.24537603557109833
+      - -0.7267504930496216
+      - -0.0816086083650589
+      - -0.17939254641532898
+      - 0.1138397604227066
+      - -0.8360004425048828
+      - 0.47255510091781616
+      - -0.07358624786138535
+      - 0.1038229838013649
+      - -0.07785441726446152
+      - 0.7885988354682922
+      - -1.165297269821167
+      - -0.2916831970214844
+      - 0.5662569403648376
+      - -0.6377986669540405
+      - -0.05221424624323845
+      - -0.6274069547653198
+      - 0.13630224764347076
+      - -0.29677602648735046
+      - -0.6062723398208618
+      - -0.2247072458267212
+      - 0.35355159640312195
+      - 0.36897560954093933
+      - 0.27751320600509644
+      - -0.14971300959587097
+      - -0.19294434785842896
+      - -0.1479526162147522
+      - -0.02374897710978985
+      - -0.14040935039520264
+      - 0.25790536403656006
+      - 0.16871976852416992
+      - -0.06627867370843887
+    target_std:
+      - 2.350555181503296
+      - 1.936501145362854
+      - 1.0859044790267944
+      - 3.1779537200927734
+      - 1.4753531217575073
+      - 2.520535707473755
+      - 2.101593017578125
+      - 3.3897223472595215
+      - 1.3110147714614868
+      - 1.770803451538086
+      - 1.8564962148666382
+      - 2.5058887004852295
+      - 2.455212354660034
+      - 2.7981133460998535
+      - 2.7846906185150146
+      - 2.51442289352417
+      - 2.157825469970703
+      - 1.4927104711532593
+      - 2.9518864154815674
+      - 1.0375343561172485
+      - 3.0358963012695312
+      - 2.821928024291992
+      - 1.0281389951705933
+      - 2.0190064907073975
+      - 2.2814815044403076
+      - 2.8889224529266357
+      - 0.8938020467758179
+      - 2.296116352081299
+      - 2.3712480068206787
+      - 1.9524204730987549
+      - 2.948482036590576
+      - 1.6851431131362915

ckpts/sqvae_geomae.safetensors

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

ckpts/sqvae_geomae.yaml

@@ -0,0 +1,28 @@
+model:
+  params:
+    latent_vae_params:
+      in_keys_list: ["dcdf", "gdcdf", "cdf", "gcdf", "normal", "offset", "offset1", "offset2", "offset3", "offsetb", "offsetc", "offsetd"]
+      out_keys_list: ["dcdf", "cdf", "offset", "offset1", "offset2", "offset3", "offsetb", "offsetc", "offsetd"]
+      out_weights_list: [1, 1, 1, 1, 1, 1, 1, 1, 1]
+      dim: 512
+      num_latents: 1024
+      latent_dim: 32
+      attn_depth: 24
+      loss_params:
+        kl_weight: 0.001
+      is_learnable_latents: 0
+      is_context_norm: 0
+      use_geom_code_as_fps: 1
+      use_geom_code_and_pe_as_fps: 1
+      gcolor_grad_norm_in: 1
+
+    cond_vae_params:
+      in_keys_list: ["normal"]
+      out_keys_list: ["udf"]
+      out_weights_list: [1]
+      dim: 512
+      num_latents: 1024
+      latent_dim: 64
+      attn_depth: 24
+      loss_params:
+        kl_weight: 0.0