| { | |
| "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" | |
| } |