Skip to content

Commit

Permalink
Update modelcard.md
Browse files Browse the repository at this point in the history
  • Loading branch information
@hardmaru
hardmaru authored Dec 7, 2022
1 parent 64888bc commit 2fc5104
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions modelcard.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -94,8 +94,8 @@ We currently provide the following checkpoints, for various versions:

### Version 2.1

`512-base-ema.ckpt`: Fine-tuned on `512-base-ema.ckpt` 2.0 with 220k extra steps taken, with `punsafe=0.98` on the same dataset.
`768-v-ema.ckpt`: Resumed from `768-v-ema.ckpt` 2.0 with an additional 55k steps on the same dataset (`punsafe=0.1`), and then fine-tuned for another 155k extra steps with `punsafe=0.98`.
- `512-base-ema.ckpt`: Fine-tuned on `512-base-ema.ckpt` 2.0 with 220k extra steps taken, with `punsafe=0.98` on the same dataset.
- `768-v-ema.ckpt`: Resumed from `768-v-ema.ckpt` 2.0 with an additional 55k steps on the same dataset (`punsafe=0.1`), and then fine-tuned for another 155k extra steps with `punsafe=0.98`.
### Version 2.0

- `512-base-ema.ckpt`: 550k steps at resolution `256x256` on a subset of [LAION-5B](https://laion.ai/blog/laion-5b/) filtered for explicit pornographic material, using the [LAION-NSFW classifier](https://github.com/LAION-AI/CLIP-based-NSFW-Detector) with `punsafe=0.1` and an [aesthetic score](https://github.com/christophschuhmann/improved-aesthetic-predictor) >= `4.5`.
Expand Down

0 comments on commit 2fc5104

Please sign in to comment.