From 2fc510429ce55ad2b4cbbb53f400326bf8a8d19c Mon Sep 17 00:00:00 2001 From: hardmaru Date: Wed, 7 Dec 2022 16:31:01 +0900 Subject: [PATCH] Update modelcard.md --- modelcard.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/modelcard.md b/modelcard.md index daa328e..787f15c 100644 --- a/modelcard.md +++ b/modelcard.md @@ -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`.