update renderer, remove params_proj

model_index.json

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

params_proj/config.json

@@ -1,29 +0,0 @@
-{
-  "_class_name": "ShapEParamsProjModel",
-  "_diffusers_version": "0.17.0.dev0",
-  "d_latent": 1024,
-  "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
-    ]
-  ]
-}

params_proj/diffusion_pytorch_model.bin

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

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