up

config.json

@@ -1,14 +1,17 @@
 {
-  "_model_class": "UNetModel",
-  "channels": 3,
-  "dim": 8,
-  "dim_mults": [
+  "_class_name": "UNetModel",
+  "attn_resolutions": [
+    16
+  ],
+  "ch": 32,
+  "ch_mult": [
     1,
     2
   ],
-  "init_dim": null,
-  "learned_variance": false,
-  "out_dim": null,
-  "resnet_block_groups": 2,
-  "with_time_emb": true
+  "dropout": 0.0,
+  "in_channels": 3,
+  "num_res_blocks": 2,
+  "out_ch": 3,
+  "resamp_with_conv": true,
+  "resolution": 32
 }

diffusion_model.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d284c6f43acade4655fcc7042d8974074bb66c5a8090ef61e9076c06da9e879c
-size 286841
+oid sha256:3ef275c749edcecd52e9b4a6215591b9516417ba77f4c5e595a3f9a7d3caf776
+size 4079329

sampler_config.json

@@ -1,9 +0,0 @@
-{
-  "_class_name": "GaussianDiffusion",
-  "beta_schedule": "cosine",
-  "channels": 3,
-  "image_size": 128,
-  "loss_type": "l1",
-  "objective": "pred_noise",
-  "timesteps": 3
-}