Update config.json
Browse files- config.json +4 -20
config.json
CHANGED
|
@@ -1,26 +1,10 @@
|
|
| 1 |
{
|
| 2 |
"model_name": "JPG_Noise_Remover",
|
| 3 |
-
"architectures":
|
| 4 |
-
"legacy_v1": "ResUNet (v1)",
|
| 5 |
-
"nafnet_v2": "NAFNet UNet (v2)"
|
| 6 |
-
},
|
| 7 |
"files": [
|
| 8 |
-
{
|
| 9 |
-
|
| 10 |
-
|
| 11 |
-
"base_ch": 64
|
| 12 |
-
},
|
| 13 |
-
{
|
| 14 |
-
"file": "best_ema_v2_E11_BF16.safetensors",
|
| 15 |
-
"arch": "nafnet_v2",
|
| 16 |
-
"base_ch": 64,
|
| 17 |
-
"amp": "bf16"
|
| 18 |
-
},
|
| 19 |
-
{
|
| 20 |
-
"file": "best_ema_15E.safetensors",
|
| 21 |
-
"arch": "legacy_v1",
|
| 22 |
-
"base_ch": 64
|
| 23 |
-
}
|
| 24 |
],
|
| 25 |
"default": "best_ema_v2_E11.pt",
|
| 26 |
"task": "image-to-image",
|
|
|
|
| 1 |
{
|
| 2 |
"model_name": "JPG_Noise_Remover",
|
| 3 |
+
"architectures": ["legacy_v1", "nafnet_v2"], // список, не объект
|
|
|
|
|
|
|
|
|
|
| 4 |
"files": [
|
| 5 |
+
{ "file": "best_ema_v2_E11.pt", "arch": "nafnet_v2", "base_ch": 64 },
|
| 6 |
+
{ "file": "best_ema_v2_E11_BF16.safetensors", "arch": "nafnet_v2", "base_ch": 64, "amp": "bf16" },
|
| 7 |
+
{ "file": "best_ema_15E.safetensors", "arch": "legacy_v1", "base_ch": 64 }
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
],
|
| 9 |
"default": "best_ema_v2_E11.pt",
|
| 10 |
"task": "image-to-image",
|