update

model_index.json

@@ -16,5 +16,13 @@
   "prior": [
     "diffusers",
     "PriorTransformer"
+  ],
+  "params_proj": [
+    "shap_e",
+    "ShapEParamsProjModel"
+  ],
+  "renderer": [
+    "shap_e",
+    "MLPNeRSTFModel"
   ]
 }

params_proj/config.json

@@ -0,0 +1,29 @@
+{
+  "_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

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

renderer/config.json

@@ -0,0 +1,9 @@
+{
+  "_class_name": "MLPNeRSTFModel",
+  "_diffusers_version": "0.17.0.dev0",
+  "act_fn": "swish",
+  "d_hidden": 256,
+  "insert_direction_at": 4,
+  "n_hidden_layers": 6,
+  "n_output": 12
+}

renderer/diffusion_pytorch_model.bin

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

scheduler/backup_scheduler_config.json

@@ -1,182 +0,0 @@
-{
-  "_class_name": "HeunDiscreteScheduler",
-  "_diffusers_version": "0.17.0.dev0",
-  "beta_end": 2.3703e-02,
-  "beta_schedule": "scaled_linear",
-  "beta_start": 9.5367e-07,
-  "trained_betas": [
-    9.5367e-07, 1.1921e-07, 0.0000e+00, 0.0000e+00, 0.0000e+00, 0.0000e+00,
-    1.1921e-07, 0.0000e+00, 0.0000e+00, 0.0000e+00, 1.1921e-07, 0.0000e+00, 
-    0.0000e+00, 0.0000e+00, 1.1921e-07, 0.0000e+00, 0.0000e+00, 1.1921e-07,
-    0.0000e+00, 0.0000e+00, 1.1921e-07, 0.0000e+00, 0.0000e+00, 1.1921e-07,
-    0.0000e+00, 0.0000e+00, 1.1921e-07, 0.0000e+00, 0.0000e+00, 1.1921e-07,
-    0.0000e+00, 1.1921e-07, 0.0000e+00, 1.1921e-07, 0.0000e+00, 0.0000e+00,
-    1.1921e-07, 0.0000e+00, 1.1921e-07, 0.0000e+00, 1.1921e-07, 0.0000e+00,
-    1.1921e-07, 1.1921e-07, 0.0000e+00, 1.1921e-07, 0.0000e+00, 1.1921e-07,
-    1.1921e-07, 0.0000e+00, 1.1921e-07, 1.1921e-07, 0.0000e+00, 1.1921e-07,
-    1.1921e-07, 0.0000e+00, 1.1921e-07, 1.1921e-07, 1.1921e-07, 1.1921e-07,
-    0.0000e+00, 1.1921e-07, 1.1921e-07, 1.1921e-07, 1.1921e-07, 1.1921e-07,
-    1.1921e-07, 1.1921e-07, 1.1921e-07, 1.1921e-07, 1.1921e-07, 1.1921e-07,
-    1.1921e-07, 1.1921e-07, 2.3842e-07, 1.1921e-07, 1.1921e-07, 1.1921e-07,
-    1.1921e-07, 2.3842e-07, 1.1921e-07, 2.3842e-07, 1.1921e-07, 1.1921e-07,
-    2.3842e-07, 1.1921e-07, 2.3842e-07, 2.3842e-07, 1.1921e-07, 2.3842e-07,
-    2.3842e-07, 1.1921e-07, 2.3842e-07, 2.3842e-07, 2.3842e-07, 2.3842e-07,
-    2.3842e-07, 2.3842e-07, 2.3842e-07, 2.3842e-07, 2.3842e-07, 3.5763e-07,
-    2.3842e-07, 2.3842e-07, 3.5763e-07, 2.3842e-07, 3.5763e-07, 2.3842e-07,
-    3.5763e-07, 3.5763e-07, 2.3842e-07, 3.5763e-07, 3.5763e-07, 3.5763e-07,
-    3.5763e-07, 3.5763e-07, 4.7684e-07, 3.5763e-07, 3.5763e-07, 4.7684e-07,
-    3.5763e-07, 4.7684e-07, 4.7684e-07, 3.5763e-07, 4.7684e-07, 4.7684e-07,
-    4.7684e-07, 5.9605e-07, 4.7684e-07, 4.7684e-07, 5.9605e-07, 4.7684e-07,
-    5.9605e-07, 5.9605e-07, 5.9605e-07, 5.9605e-07, 5.9605e-07, 5.9605e-07,
-    7.1526e-07, 5.9605e-07, 7.1526e-07, 7.1526e-07, 7.1526e-07, 7.1526e-07,
-    7.1526e-07, 8.3447e-07, 7.1526e-07, 8.3447e-07, 8.3447e-07, 8.3447e-07,
-    9.5367e-07, 8.3447e-07, 9.5367e-07, 8.3447e-07, 9.5367e-07, 1.0729e-06,
-    9.5367e-07, 1.0729e-06, 9.5367e-07, 1.0729e-06, 1.1921e-06, 1.0729e-06,
-    1.1921e-06, 1.1921e-06, 1.1921e-06, 1.1921e-06, 1.3113e-06, 1.3113e-06,
-    1.3113e-06, 1.4305e-06, 1.3113e-06, 1.4305e-06, 1.5497e-06, 1.4305e-06,
-    1.5497e-06, 1.5497e-06, 1.6689e-06, 1.6689e-06, 1.6689e-06, 1.7881e-06,
-    1.7881e-06, 1.7881e-06, 1.9073e-06, 1.9073e-06, 1.9073e-06, 2.0266e-06,
-    2.0266e-06, 2.1458e-06, 2.1458e-06, 2.1458e-06, 2.2650e-06, 2.3842e-06,
-    2.3842e-06, 2.3842e-06, 2.5034e-06, 2.5034e-06, 2.6226e-06, 2.7418e-06,
-    2.7418e-06, 2.7418e-06, 2.8610e-06, 2.9802e-06, 2.9802e-06, 3.0994e-06,
-    3.2187e-06, 3.2187e-06, 3.3379e-06, 3.3379e-06, 3.4571e-06, 3.5763e-06,
-    3.6955e-06, 3.6955e-06, 3.9339e-06, 3.9339e-06, 3.9339e-06, 4.1127e-06,
-    4.1723e-06, 4.2915e-06, 4.5300e-06, 4.5300e-06, 4.6492e-06, 4.7684e-06,
-    4.7684e-06, 5.0068e-06, 5.1260e-06, 5.2452e-06, 5.3644e-06, 5.4836e-06,
-    5.6028e-06, 5.8413e-06, 5.8413e-06, 6.0797e-06, 6.1989e-06, 6.3181e-06,
-    6.5565e-06, 6.5565e-06, 6.9141e-06, 6.9141e-06, 7.0930e-06, 7.3910e-06,
-    7.5102e-06, 7.6294e-06, 7.8678e-06, 8.1062e-06, 8.2254e-06, 8.4639e-06,
-    8.7023e-06, 8.8215e-06, 9.0599e-06, 9.3579e-06, 9.5367e-06, 9.7752e-06,
-    1.0014e-05, 1.0252e-05, 1.0490e-05, 1.0729e-05, 1.0908e-05, 1.1325e-05,
-    1.1563e-05, 1.1802e-05, 1.2159e-05, 1.2338e-05, 1.2755e-05, 1.2994e-05,
-    1.3351e-05, 1.3649e-05, 1.3947e-05, 1.4305e-05, 1.4782e-05, 1.4961e-05,
-    1.5378e-05, 1.5855e-05, 1.6034e-05, 1.6570e-05, 1.6868e-05, 1.7405e-05,
-    1.7881e-05, 1.8060e-05, 1.8716e-05, 1.9133e-05, 1.9550e-05, 1.9968e-05,
-    2.0564e-05, 2.0981e-05, 2.1517e-05, 2.1994e-05, 2.2650e-05, 2.3186e-05,
-    2.3663e-05, 2.4259e-05, 2.4915e-05, 2.5451e-05, 2.6047e-05, 2.6643e-05,
-    2.7418e-05, 2.8074e-05, 2.8729e-05, 2.9445e-05, 3.0160e-05, 3.0875e-05,
-    3.1650e-05, 3.2425e-05, 3.3081e-05, 3.4034e-05, 3.4809e-05, 3.5584e-05,
-    3.6538e-05, 3.7372e-05, 3.8326e-05, 3.9160e-05, 4.0233e-05, 4.1127e-05,
-    4.2140e-05, 4.3035e-05, 4.4286e-05, 4.5180e-05, 4.6372e-05, 4.7505e-05,
-    4.8637e-05, 4.9889e-05, 5.1022e-05, 5.2333e-05, 5.3525e-05, 5.4836e-05,
-    5.6148e-05, 5.7518e-05, 5.8830e-05, 6.0439e-05, 6.1870e-05, 6.3241e-05,
-    6.4790e-05, 6.6459e-05, 6.7949e-05, 6.9678e-05, 7.1347e-05, 7.3075e-05,
-    7.4863e-05, 7.6652e-05, 7.8499e-05, 8.0407e-05, 8.2374e-05, 8.4341e-05,
-    8.6308e-05, 8.8453e-05, 9.0659e-05, 9.2864e-05, 9.4950e-05, 9.7394e-05,
-    9.9719e-05, 1.0204e-04, 1.0461e-04, 1.0717e-04, 1.0961e-04, 1.1235e-04,
-    1.1510e-04, 1.1790e-04, 1.2064e-04, 1.2356e-04, 1.2666e-04, 1.2970e-04,
-    1.3280e-04, 1.3596e-04, 1.3924e-04, 1.4263e-04, 1.4597e-04, 1.4961e-04,
-    1.5324e-04, 1.5676e-04, 1.6069e-04, 1.6457e-04, 1.6856e-04, 1.7256e-04,
-    1.7679e-04, 1.8102e-04, 1.8537e-04, 1.8984e-04, 1.9431e-04, 1.9920e-04,
-    2.0373e-04, 2.0880e-04, 2.1386e-04, 2.1899e-04, 2.2411e-04, 2.2960e-04,
-    2.3508e-04, 2.4074e-04, 2.4652e-04, 2.5249e-04, 2.5845e-04, 2.6476e-04,
-    2.7114e-04, 2.7758e-04, 2.8419e-04, 2.9111e-04, 2.9802e-04, 3.0524e-04,
-    3.1239e-04, 3.2002e-04, 3.2765e-04, 3.3551e-04, 3.4356e-04, 3.5173e-04,
-    3.6019e-04, 3.6877e-04, 3.7766e-04, 3.8648e-04, 3.9583e-04, 4.0531e-04,
-    4.1485e-04, 4.2486e-04, 4.3499e-04, 4.4531e-04, 4.5586e-04, 4.6676e-04,
-    4.7797e-04, 4.8918e-04, 5.0080e-04, 5.1284e-04, 5.2488e-04, 5.3734e-04,
-    5.5021e-04, 5.6314e-04, 5.7656e-04, 5.9015e-04, 6.0409e-04, 6.1840e-04,
-    6.3306e-04, 6.4790e-04, 6.6328e-04, 6.7890e-04, 6.9487e-04, 7.1138e-04,
-    7.2795e-04, 7.4518e-04, 7.6258e-04, 7.8058e-04, 7.9882e-04, 8.1754e-04,
-    8.3673e-04, 8.5634e-04, 8.7631e-04, 8.9681e-04, 9.1779e-04, 9.3913e-04,
-    9.6112e-04, 9.8336e-04, 1.0063e-03, 1.0296e-03, 1.0536e-03, 1.0779e-03,
-    1.1029e-03, 1.1284e-03, 1.1544e-03, 1.1811e-03, 1.2084e-03, 1.2360e-03,
-    1.2646e-03, 1.2935e-03, 1.3232e-03, 1.3536e-03, 1.3844e-03, 1.4161e-03,
-    1.4483e-03, 1.4814e-03, 1.5149e-03, 1.5494e-03, 1.5844e-03, 1.6203e-03,
-    1.6569e-03, 1.6942e-03, 1.7323e-03, 1.7713e-03, 1.8110e-03, 1.8515e-03,
-    1.8928e-03, 1.9351e-03, 1.9783e-03, 2.0220e-03, 2.0669e-03, 2.1127e-03,
-    2.1593e-03, 2.2068e-03, 2.2553e-03, 2.3048e-03, 2.3551e-03, 2.4065e-03,
-    2.4588e-03, 2.5123e-03, 2.5666e-03, 2.6221e-03, 2.6785e-03, 2.7361e-03,
-    2.7946e-03, 2.8543e-03, 2.9151e-03, 2.9770e-03, 3.0399e-03, 3.1041e-03,
-    3.1696e-03, 3.2359e-03, 3.3035e-03, 3.3723e-03, 3.4423e-03, 3.5135e-03,
-    3.5861e-03, 3.6595e-03, 3.7343e-03, 3.8105e-03, 3.8878e-03, 3.9664e-03,
-    4.0463e-03, 4.1274e-03, 4.2099e-03, 4.2935e-03, 4.3784e-03, 4.4649e-03,
-    4.5524e-03, 4.6414e-03, 4.7315e-03, 4.8230e-03, 4.9159e-03, 5.0099e-03,
-    5.1054e-03, 5.2021e-03, 5.3002e-03, 5.3996e-03, 5.5003e-03, 5.6022e-03,
-    5.7055e-03, 5.8101e-03, 5.9161e-03, 6.0230e-03, 6.1316e-03, 6.2412e-03,
-    6.3521e-03, 6.4644e-03, 6.5776e-03, 6.6923e-03, 6.8081e-03, 6.9251e-03,
-    7.0434e-03, 7.1626e-03, 7.2832e-03, 7.4047e-03, 7.5275e-03, 7.6512e-03,
-    7.7761e-03, 7.9019e-03, 8.0287e-03, 8.1567e-03, 8.2855e-03, 8.4152e-03,
-    8.5459e-03, 8.6774e-03, 8.8098e-03, 8.9430e-03, 9.0770e-03, 9.2117e-03,
-    9.3471e-03, 9.4834e-03, 9.6201e-03, 9.7576e-03, 9.8956e-03, 1.0034e-02,
-    1.0173e-02, 1.0313e-02, 1.0453e-02, 1.0593e-02, 1.0734e-02, 1.0875e-02,
-    1.1016e-02, 1.1158e-02, 1.1300e-02, 1.1441e-02, 1.1584e-02, 1.1726e-02,
-    1.1868e-02, 1.2010e-02, 1.2152e-02, 1.2294e-02, 1.2436e-02, 1.2578e-02,
-    1.2719e-02, 1.2861e-02, 1.3002e-02, 1.3142e-02, 1.3282e-02, 1.3422e-02,
-    1.3562e-02, 1.3701e-02, 1.3839e-02, 1.3977e-02, 1.4114e-02, 1.4251e-02,
-    1.4387e-02, 1.4522e-02, 1.4657e-02, 1.4791e-02, 1.4924e-02, 1.5056e-02,
-    1.5187e-02, 1.5318e-02, 1.5447e-02, 1.5576e-02, 1.5704e-02, 1.5830e-02,
-    1.5956e-02, 1.6081e-02, 1.6204e-02, 1.6326e-02, 1.6448e-02, 1.6568e-02,
-    1.6687e-02, 1.6805e-02, 1.6922e-02, 1.7037e-02, 1.7152e-02, 1.7265e-02,
-    1.7377e-02, 1.7487e-02, 1.7597e-02, 1.7705e-02, 1.7812e-02, 1.7917e-02,
-    1.8022e-02, 1.8125e-02, 1.8226e-02, 1.8327e-02, 1.8426e-02, 1.8524e-02,
-    1.8620e-02, 1.8715e-02, 1.8809e-02, 1.8902e-02, 1.8993e-02, 1.9083e-02,
-    1.9171e-02, 1.9259e-02, 1.9345e-02, 1.9429e-02, 1.9513e-02, 1.9595e-02,
-    1.9676e-02, 1.9755e-02, 1.9834e-02, 1.9911e-02, 1.9987e-02, 2.0061e-02,
-    2.0135e-02, 2.0207e-02, 2.0278e-02, 2.0347e-02, 2.0416e-02, 2.0483e-02,
-    2.0549e-02, 2.0614e-02, 2.0678e-02, 2.0741e-02, 2.0803e-02, 2.0863e-02,
-    2.0923e-02, 2.0981e-02, 2.1038e-02, 2.1095e-02, 2.1150e-02, 2.1204e-02,
-    2.1257e-02, 2.1309e-02, 2.1360e-02, 2.1411e-02, 2.1460e-02, 2.1508e-02,
-    2.1555e-02, 2.1602e-02, 2.1647e-02, 2.1692e-02, 2.1736e-02, 2.1779e-02,
-    2.1820e-02, 2.1862e-02, 2.1902e-02, 2.1942e-02, 2.1980e-02, 2.2018e-02,
-    2.2056e-02, 2.2092e-02, 2.2128e-02, 2.2163e-02, 2.2197e-02, 2.2230e-02,
-    2.2263e-02, 2.2295e-02, 2.2327e-02, 2.2358e-02, 2.2388e-02, 2.2417e-02,
-    2.2446e-02, 2.2475e-02, 2.2502e-02, 2.2529e-02, 2.2556e-02, 2.2582e-02,
-    2.2607e-02, 2.2632e-02, 2.2656e-02, 2.2680e-02, 2.2703e-02, 2.2726e-02,
-    2.2748e-02, 2.2770e-02, 2.2791e-02, 2.2812e-02, 2.2833e-02, 2.2852e-02,
-    2.2872e-02, 2.2891e-02, 2.2910e-02, 2.2928e-02, 2.2946e-02, 2.2963e-02,
-    2.2980e-02, 2.2997e-02, 2.3013e-02, 2.3029e-02, 2.3045e-02, 2.3060e-02,
-    2.3074e-02, 2.3089e-02, 2.3103e-02, 2.3117e-02, 2.3131e-02, 2.3144e-02,
-    2.3157e-02, 2.3170e-02, 2.3182e-02, 2.3194e-02, 2.3206e-02, 2.3218e-02,
-    2.3229e-02, 2.3240e-02, 2.3251e-02, 2.3261e-02, 2.3272e-02, 2.3282e-02,
-    2.3291e-02, 2.3302e-02, 2.3310e-02, 2.3320e-02, 2.3329e-02, 2.3337e-02,
-    2.3347e-02, 2.3354e-02, 2.3363e-02, 2.3371e-02, 2.3378e-02, 2.3386e-02,
-    2.3393e-02, 2.3401e-02, 2.3408e-02, 2.3415e-02, 2.3421e-02, 2.3429e-02,
-    2.3434e-02, 2.3441e-02, 2.3447e-02, 2.3453e-02, 2.3459e-02, 2.3465e-02,
-    2.3471e-02, 2.3476e-02, 2.3481e-02, 2.3487e-02, 2.3491e-02, 2.3497e-02,
-    2.3502e-02, 2.3506e-02, 2.3511e-02, 2.3515e-02, 2.3520e-02, 2.3525e-02,
-    2.3529e-02, 2.3533e-02, 2.3537e-02, 2.3540e-02, 2.3544e-02, 2.3549e-02,
-    2.3552e-02, 2.3556e-02, 2.3559e-02, 2.3562e-02, 2.3566e-02, 2.3569e-02,
-    2.3573e-02, 2.3575e-02, 2.3578e-02, 2.3582e-02, 2.3584e-02, 2.3587e-02,
-    2.3590e-02, 2.3592e-02, 2.3595e-02, 2.3598e-02, 2.3600e-02, 2.3603e-02,
-    2.3605e-02, 2.3607e-02, 2.3610e-02, 2.3612e-02, 2.3614e-02, 2.3616e-02,
-    2.3618e-02, 2.3621e-02, 2.3622e-02, 2.3624e-02, 2.3627e-02, 2.3628e-02,
-    2.3630e-02, 2.3632e-02, 2.3633e-02, 2.3635e-02, 2.3636e-02, 2.3638e-02,
-    2.3640e-02, 2.3641e-02, 2.3643e-02, 2.3644e-02, 2.3646e-02, 2.3647e-02,
-    2.3649e-02, 2.3650e-02, 2.3651e-02, 2.3652e-02, 2.3653e-02, 2.3655e-02,
-    2.3656e-02, 2.3658e-02, 2.3658e-02, 2.3659e-02, 2.3661e-02, 2.3661e-02,
-    2.3662e-02, 2.3664e-02, 2.3664e-02, 2.3666e-02, 2.3666e-02, 2.3667e-02,
-    2.3668e-02, 2.3669e-02, 2.3670e-02, 2.3670e-02, 2.3671e-02, 2.3672e-02,
-    2.3673e-02, 2.3674e-02, 2.3675e-02, 2.3675e-02, 2.3676e-02, 2.3677e-02,
-    2.3677e-02, 2.3678e-02, 2.3678e-02, 2.3679e-02, 2.3680e-02, 2.3680e-02,
-    2.3681e-02, 2.3681e-02, 2.3682e-02, 2.3683e-02, 2.3683e-02, 2.3683e-02,
-    2.3684e-02, 2.3685e-02, 2.3685e-02, 2.3686e-02, 2.3686e-02, 2.3687e-02,
-    2.3687e-02, 2.3687e-02, 2.3688e-02, 2.3688e-02, 2.3688e-02, 2.3688e-02,
-    2.3689e-02, 2.3690e-02, 2.3690e-02, 2.3690e-02, 2.3691e-02, 2.3691e-02,
-    2.3691e-02, 2.3692e-02, 2.3692e-02, 2.3692e-02, 2.3692e-02, 2.3693e-02,
-    2.3693e-02, 2.3693e-02, 2.3694e-02, 2.3694e-02, 2.3694e-02, 2.3694e-02,
-    2.3694e-02, 2.3694e-02, 2.3695e-02, 2.3695e-02, 2.3695e-02, 2.3695e-02,
-    2.3696e-02, 2.3696e-02, 2.3696e-02, 2.3696e-02, 2.3697e-02, 2.3697e-02,
-    2.3698e-02, 2.3697e-02, 2.3697e-02, 2.3697e-02, 2.3697e-02, 2.3698e-02,
-    2.3698e-02, 2.3698e-02, 2.3698e-02, 2.3698e-02, 2.3698e-02, 2.3699e-02,
-    2.3699e-02, 2.3700e-02, 2.3699e-02, 2.3699e-02, 2.3700e-02, 2.3699e-02,
-    2.3700e-02, 2.3699e-02, 2.3699e-02, 2.3701e-02, 2.3700e-02, 2.3700e-02,
-    2.3700e-02, 2.3700e-02, 2.3700e-02, 2.3700e-02, 2.3700e-02, 2.3702e-02,
-    2.3701e-02, 2.3701e-02, 2.3701e-02, 2.3701e-02, 2.3701e-02, 2.3701e-02,
-    2.3701e-02, 2.3702e-02, 2.3701e-02, 2.3701e-02, 2.3702e-02, 2.3701e-02,
-    2.3702e-02, 2.3702e-02, 2.3702e-02, 2.3703e-02, 2.3702e-02, 2.3702e-02,
-    2.3702e-02, 2.3702e-02, 2.3702e-02, 2.3702e-02, 2.3702e-02, 2.3703e-02,
-    2.3702e-02, 2.3702e-02, 2.3703e-02, 2.3702e-02, 2.3702e-02, 2.3702e-02,
-    2.3702e-02, 2.3703e-02, 2.3703e-02, 2.3703e-02, 2.3703e-02, 2.3703e-02,
-    2.3703e-02, 2.3703e-02, 2.3703e-02, 2.3704e-02, 2.3703e-02, 2.3703e-02,
-    2.3704e-02, 2.3703e-02, 2.3703e-02, 2.3703e-02, 2.3703e-02, 2.3704e-02,
-    2.3703e-02, 2.3703e-02, 2.3703e-02, 2.3703e-02, 2.3703e-02, 2.3703e-02,
-    2.3703e-02, 2.3704e-02, 2.3703e-02, 2.3703e-02],
-  "clip_sample": false,
-  "num_train_timesteps": 1000,
-  "prediction_type": "epsilon",
-  "set_alpha_to_one": false,
-  "skip_prk_steps": true,
-  "steps_offset": 1,
-  "use_karras_sigmas": true
-}