update renderer

model_index.json

@@ -17,12 +17,8 @@
     "diffusers",
     "PriorTransformer"
   ],
-  "params_proj": [
-    "shap_e",
-    "ShapEParamsProjModel"
-  ],
   "renderer": [
     "shap_e",
-    "MLPNeRSTFModel"
+    "ShapERenderer"
   ]
 }

renderer/config.json

@@ -1,9 +1,34 @@
 {
-  "_class_name": "MLPNeRSTFModel",
-  "_diffusers_version": "0.17.0.dev0",
+  "_class_name": "ShapERenderer",
+  "_diffusers_version": "0.18.0.dev0",
   "act_fn": "swish",
   "d_hidden": 256,
+  "d_latent": 1024,
   "insert_direction_at": 4,
   "n_hidden_layers": 6,
-  "n_output": 12
+  "n_output": 12,
+  "param_names": [
+    "nerstf.mlp.0.weight",
+    "nerstf.mlp.1.weight",
+    "nerstf.mlp.2.weight",
+    "nerstf.mlp.3.weight"
+  ],
+  "param_shapes": [
+    [
+      256,
+      93
+    ],
+    [
+      256,
+      256
+    ],
+    [
+      256,
+      256
+    ],
+    [
+      256,
+      256
+    ]
+  ]
 }

renderer/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6bac1ac152675a9c58140f336aaa1e255b65188035eea4cbb32be2738c9b13de
-size 1480443
+oid sha256:2b50111f786c3dfca1d14cb1536a761cbc6826923a1f0021a19e156df5e171ee
+size 905200536