Upload folder using huggingface_hub

README.md

@@ -1,3 +1,9 @@
 ---
-license: apache-2.0
+library_name: pytorch
+tags: [image-restoration, denoising, super-resolution, image-enhancement]
 ---
+# RestoreNet
+
+Restormer-style image restoration model with a task embedding.
+Tasks: denoise, upscale, enhance. Input and output are RGB images in [0, 1].
+Trained on synthetic degradations of images from `FlameF0X/nano-banana-pro-gen-zh-en`.

config.json

@@ -0,0 +1,20 @@
+{
+  "model_type": "restore_net",
+  "architectures": [
+    "RestoreNet"
+  ],
+  "base_channels": 64,
+  "blocks": [
+    4,
+    6,
+    8
+  ],
+  "emb_dim": 64,
+  "tasks": [
+    "denoise",
+    "upscale",
+    "enhance"
+  ],
+  "training_steps": 2000,
+  "training_dataset": "FlameF0X/nano-banana-pro-gen-zh-en"
+}

model.safetensors

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