From 2ecccb629005ccdd887052a43e5fe0a637d439d6 Mon Sep 17 00:00:00 2001 From: Qing Date: Thu, 20 Oct 2022 21:03:43 +0800 Subject: [PATCH] update readme --- README.md | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index d91f03d..8eefec6 100644 --- a/README.md +++ b/README.md @@ -32,7 +32,7 @@ 1. [ZITS](https://github.com/DQiaole/ZITS_inpainting) 1. [MAT](https://github.com/fenglinglwb/MAT) 1. [FcF](https://github.com/SHI-Labs/FcF-Inpainting) - 1. [SD1.4](https://github.com/CompVis/stable-diffusion) + 1. [SD1.5](https://github.com/runwayml/stable-diffusion) - Support CPU & GPU - Various inpainting [strategy](#inpainting-strategy) - Run as a desktop APP @@ -87,7 +87,7 @@ Available command line arguments: | Name | Description | Default | | -------------------- | ----------------------------------------------------------------------------------------------------------------------------- | -------- | -| --model | lama/ldm/zits/mat/fcf/sd1.4 See details in [Inpaint Model](#inpainting-model) | lama | +| --model | lama/ldm/zits/mat/fcf/sd1.5 See details in [Inpaint Model](#inpainting-model) | lama | | --hf_access_token | stable-diffusion(sd) model need [huggingface access token](https://huggingface.co/docs/hub/security-tokens) to download model | | | --sd-run-local | Once the model as downloaded, you can pass this arg and remove `--hf_access_token` | | | --sd-disable-nsfw | Disable stable-diffusion NSFW checker. | | @@ -109,7 +109,7 @@ Available command line arguments: | ZITS | :+1: Better holistic structures compared with previous methods
:neutral_face: Wireframe module is **very** slow on CPU | `Wireframe`: Enable edge and line detect | | MAT | TODO | | | FcF | :+1: Better structure and texture generation
:neutral_face: Only support fixed size (512x512) input | | -| SD1.4 | :+1: SOTA text-to-image diffusion model | | +| SD1.5 | :+1: SOTA text-to-image diffusion model | |
See model comparison detail @@ -184,7 +184,7 @@ so you don't have to download the model every time you start the container. The cache directories for different models correspond as follows: - lama/ldm/zits/mat/fcf: /root/.cache/torch -- sd1.4: /root/.cache/huggingface +- sd1.5: /root/.cache/huggingface ### Run Docker (cpu)