Update README.md
Browse files
README.md
CHANGED
|
@@ -11,7 +11,8 @@ language:
|
|
| 11 |
|
| 12 |
# Info
|
| 13 |
|
| 14 |
-
A quick and dirty [NAFNet](https://github.com/megvii-research/nafnet)-small based image restoration model for Qwen/Wan VAE grid artifact removal/mitigation, trained with 0 experience and a lot of AI handholding.
|
|
|
|
| 15 |
|
| 16 |
- **Aim:** Something on par with or hopefully better than Nyquist Notch shader - Remove the grid without changing/deforming/destroying anything
|
| 17 |
- **Limitations:** 1. Maybe realism, idk; 2. This model is only trained on what the VAE changes and very likely won't touch any extra noise or such your diffusion model generated
|
|
@@ -91,6 +92,6 @@ You can find PyTorch checkpoints as they came from training in the `train/` subf
|
|
| 91 |
- `100k` steps - pretrain
|
| 92 |
- `30k` steps - finetune
|
| 93 |
|
| 94 |
-
Training code (
|
| 95 |
|
| 96 |
<https://github.com/DraconicDragon/NAFNet-c>
|
|
|
|
| 11 |
|
| 12 |
# Info
|
| 13 |
|
| 14 |
+
A quick and dirty [NAFNet](https://github.com/megvii-research/nafnet)-small based image restoration model for Qwen/Wan VAE grid artifact removal/mitigation, trained with 0 experience and a lot of AI handholding.
|
| 15 |
+
(Not trained for Qwen 2.1/Wan 2.2 VAE, doubt it'll work but you tell me)
|
| 16 |
|
| 17 |
- **Aim:** Something on par with or hopefully better than Nyquist Notch shader - Remove the grid without changing/deforming/destroying anything
|
| 18 |
- **Limitations:** 1. Maybe realism, idk; 2. This model is only trained on what the VAE changes and very likely won't touch any extra noise or such your diffusion model generated
|
|
|
|
| 92 |
- `100k` steps - pretrain
|
| 93 |
- `30k` steps - finetune
|
| 94 |
|
| 95 |
+
Training code (configs too, probably):
|
| 96 |
|
| 97 |
<https://github.com/DraconicDragon/NAFNet-c>
|