Upload 2 files

config.json

@@ -0,0 +1,88 @@
+{
+    "module": "zea.models.diffusion",
+    "class_name": "DiffusionModel",
+    "config": {
+        "name": "diffusion_model",
+        "trainable": true,
+        "dtype": {
+            "module": "keras",
+            "class_name": "DTypePolicy",
+            "config": {
+                "name": "float32"
+            },
+            "registered_name": null
+        },
+        "input_shape": [
+            256,
+            256,
+            1
+        ],
+        "input_range": [
+            -1,
+            1
+        ],
+        "min_signal_rate": 0.02,
+        "max_signal_rate": 0.95,
+        "network_name": "unet_time_conditional",
+        "network_kwargs": {
+            "block_depth": 2,
+            "widths": [32, 64, 96, 128]
+        }
+    },
+    "registered_name": "DiffusionModel",
+    "build_config": {
+        "input_shape": [
+            [
+                1,
+                256,
+                256,
+                1
+            ],
+            [
+                1,
+                1,
+                1,
+                1
+            ]
+        ]
+    },
+    "compile_config": {
+        "optimizer": {
+            "module": "keras.optimizers",
+            "class_name": "AdamW",
+            "config": {
+                "name": "adamw",
+                "learning_rate": 9.999999747378752e-06,
+                "weight_decay": 1e-05,
+                "clipnorm": null,
+                "global_clipnorm": null,
+                "clipvalue": null,
+                "use_ema": false,
+                "ema_momentum": 0.99,
+                "ema_overwrite_frequency": null,
+                "loss_scale_factor": null,
+                "gradient_accumulation_steps": null,
+                "beta_1": 0.9,
+                "beta_2": 0.999,
+                "epsilon": 1e-07,
+                "amsgrad": false
+            },
+            "registered_name": null
+        },
+        "loss": {
+            "module": "keras.losses",
+            "class_name": "MeanSquaredError",
+            "config": {
+                "name": "mean_squared_error",
+                "reduction": "sum_over_batch_size"
+            },
+            "registered_name": null
+        },
+        "loss_weights": null,
+        "metrics": null,
+        "weighted_metrics": null,
+        "run_eagerly": false,
+        "steps_per_execution": 1,
+        "jit_compile": true
+    }
+}

model.weights.h5

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