DraconicDragon commited on
Commit
990d6d1
·
verified ·
1 Parent(s): 71f8224

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
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 (i think configs too):
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>