Skip to content

Commit

Permalink
[fix] fix disco readme (#1519)
Browse files Browse the repository at this point in the history 
Co-authored-by: liuwenran <[email protected]>
  • Loading branch information
@liuwenran
liuwenran and liuwenran authored Dec 19, 2022
1 parent 768e022 commit 1735f5c
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 16 deletions.
26 changes: 13 additions & 13 deletions configs/disco_diffusion/README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,18 +24,18 @@ Created by Somnai, augmented by Gandamu, and building on the work of RiversHaveW

We have converted several `unet` weights and offer related configs. See more details of different `unet` in [Tutorial](#tutorials).

| Diffusion Model | Config | Download |
| :--------------------------------------: | :-------------------------------------------------------------------------: | :-----------------------------------------------------------------------------------: |
| 512x512_diffusion_uncond_finetune_008100 | [config](configs/disco/disco-diffusion_adm-u-finetuned_imagenet-512x512.py) | [model](https://download.openmmlab.com/mmediting/synthesizers/disco/adm-u_finetuned_imagenet-512x512-ab471d70.pth) |
| 256x256_diffusion_uncond | [config](configs/disco/disco-diffusion_adm-u-finetuned_imagenet-256x256.py) | [model](<>) |
| portrait_generator_v001 | [config](configs/disco/disco-diffusion_portrait-generator-v001.py) | [model](https://download.openmmlab.com/mmediting/synthesizers/disco/adm-u-cvt-rgb_portrait-v001-f4a3f3bc.pth) |
| pixelartdiffusion_expanded | Coming soon! | |
| pixel_art_diffusion_hard_256 | Coming soon! | |
| pixel_art_diffusion_soft_256 | Coming soon! | |
| pixelartdiffusion4k | Coming soon! | |
| watercolordiffusion_2 | Coming soon! | |
| watercolordiffusion | Coming soon! | |
| PulpSciFiDiffusion | Coming soon! | |
| Diffusion Model | Config | Download |
| :--------------------------------------: | :-----------------------------------------------------------------------------: | :-------------------------------------------------------------------------------: |
| 512x512_diffusion_uncond_finetune_008100 | [config](configs/disco_diffusion/disco-diffusion_adm-u-finetuned_imagenet-512x512.py) | [model](https://download.openmmlab.com/mmediting/synthesizers/disco/adm-u_finetuned_imagenet-512x512-ab471d70.pth) |
| 256x256_diffusion_uncond | [config](configs/disco_diffusion/disco-diffusion_adm-u-finetuned_imagenet-256x256.py) | [model](<>) |
| portrait_generator_v001 | [config](configs/disco_diffusion/disco-diffusion_portrait-generator-v001.py) | [model](https://download.openmmlab.com/mmediting/synthesizers/disco/adm-u-cvt-rgb_portrait-v001-f4a3f3bc.pth) |
| pixelartdiffusion_expanded | Coming soon! | |
| pixel_art_diffusion_hard_256 | Coming soon! | |
| pixel_art_diffusion_soft_256 | Coming soon! | |
| pixelartdiffusion4k | Coming soon! | |
| watercolordiffusion_2 | Coming soon! | |
| watercolordiffusion | Coming soon! | |
| PulpSciFiDiffusion | Coming soon! | |

## To-do List

Expand All @@ -61,7 +61,7 @@ from torchvision.utils import save_image
register_all_modules()

disco = MODELS.build(
Config.fromfile('configs/disco/disco-baseline.py').model).cuda().eval()
Config.fromfile('configs/disco_diffusion/disco-baseline.py').model).cuda().eval()
text_prompts = {
0: [
"A beautiful painting of a singular lighthouse, shining its light across a tumultuous sea of blood by greg rutkowski and thomas kinkade, Trending on artstation.",
Expand Down
6 changes: 3 additions & 3 deletions configs/disco_diffusion/metafile.yml
View file
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ Collections:
- https://github.com/alembics/disco-diffusion
README: configs/disco_diffusion/README.md
Models:
- Config: configs/disco/disco-diffusion_adm-u-finetuned_imagenet-512x512.py
- Config: configs/disco_diffusion/disco-diffusion_adm-u-finetuned_imagenet-512x512.py
In Collection: Disco Diffusion
Metadata:
Training Data: Others
Expand All @@ -17,7 +17,7 @@ Models:
Metrics: {}
Task: Text2Image, Image2Image
Weights: https://download.openmmlab.com/mmediting/synthesizers/disco/adm-u_finetuned_imagenet-512x512-ab471d70.pth
- Config: configs/disco/disco-diffusion_adm-u-finetuned_imagenet-256x256.py
- Config: configs/disco_diffusion/disco-diffusion_adm-u-finetuned_imagenet-256x256.py
In Collection: Disco Diffusion
Metadata:
Training Data: Others
Expand All @@ -27,7 +27,7 @@ Models:
Metrics: {}
Task: Text2Image, Image2Image
Weights: <>
- Config: configs/disco/disco-diffusion_portrait-generator-v001.py
- Config: configs/disco_diffusion/disco-diffusion_portrait-generator-v001.py
In Collection: Disco Diffusion
Metadata:
Training Data: Others
Expand Down

0 comments on commit 1735f5c

Please sign in to comment.