Update README

README.md

@@ -14,7 +14,7 @@ tags:
 English version is [here](README_en.md).
 
 # はじめに
-Cool Japan Diffusion (for learning) はStable Diffsionをファインチューニングして、アニメやマンガ、ゲームなどのクールジャパンを表現することに特化したモデルです。なお、内閣府のクールジャパン戦略とは特に関係はありません。
+Cool Japan Diffusion はStable Diffsionをファインチューニングして、アニメやマンガ、ゲームなどのクールジャパンを表現することに特化したモデルです。なお、内閣府のクールジャパン戦略とは特に関係はありません。
 
 # ライセンスについて
 ライセンスについては、もとのライセンス CreativeML Open RAIL++-M License に例外を除き商用利用禁止を追加しただけです。
@@ -38,9 +38,9 @@ Cool Japan Diffusion (for learning) はStable Diffsionをファインチュー
 したがって、法的な側面ではなく、倫理的な側面を調査する目的も本配布は兼ねていると考えてください。
 
 # 使い方
-手軽に楽しみたい方は、こちらの[Space](https://huggingface.co/spaces/alfredplpl/cool-japan-diffusion-2-1-0)をお使いください。
-詳しい本モデルの取り扱い方は[こちらの取扱説明書](https://alfredplpl.hatenablog.com/entry/2022/12/30/102636)にかかれています。
-モデルは[ここ](https://huggingface.co/aipicasso/cool-japan-diffusion-2-1-0/resolve/main/v2-1-0.ckpt)からダウンロードできます。
+手軽に楽しみたい方は、こちらの[Space](https://huggingface.co/spaces/aipicasso/cool-japan-diffusion-latest-demo)をお使いください。
+詳しい本モデルの取り扱い方は[こちらの取扱説明書](https://alfredplpl.hatenablog.com/entry/2023/01/11/182146)にかかれています。
+モデルは[ここ](https://huggingface.co/aipicasso/cool-japan-diffusion-2-1-1/resolve/main/v2-1-1.ckpt)からダウンロードできます。
 
 以下、一般的なモデルカードの日本語訳です。
 
@@ -71,7 +71,7 @@ Stable Diffusion v2と同じ使い方です。
 
 ### Web UIの場合
 
-こちらの[取扱説明書](https://alfredplpl.hatenablog.com/entry/2022/12/30/102636)に従って作成してください。
+こちらの[取扱説明書](https://alfredplpl.hatenablog.com/entry/2023/01/11/182146)に従って作成してください。
 
 ### Diffusersの場合
 
@@ -86,20 +86,19 @@ pip install --upgrade git+https://github.com/huggingface/diffusers.git transform
 次のスクリプトを実行し、画像を生成してください。
 
 ```python
-from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
 import torch
 
-model_id = "aipicasso/cool-japan-diffusion-2-1-1-beta"
+model_id = "aipicasso/cool-japan-diffusion-2-1-1"
 
-scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
+scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)#,use_auth_token="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")
 pipe = pipe.to("cuda")
 
-prompt = "anime, a portrait of a girl with black short hair and red eyes, kimono, full color illustration, official art, 4k, detailed"
-negative_prompt="low quality, bad face, bad anatomy, bad hand, lowres, jpeg artifacts, 2d, 3d, cg, text"
-image = pipe(prompt,negative_prompt=negative_prompt).images[0]
-
-image.save("girl.png")
+prompt = "anime, masterpiece, a portrait of a girl, good pupil, 4k, detailed"
+negative_prompt="deformed, blurry, bad anatomy, bad pupil, disfigured, poorly drawn face, mutation, mutated, extra limb, ugly, poorly drawn hands, bad hands, fused fingers, messy drawing, broken legs censor, low quality, mutated hands and fingers, long body, mutation, poorly drawn, bad eyes, ui, error, missing fingers, fused fingers, one hand with more than 5 fingers, one hand with less than 5 fingers, one hand with more than 5 digit, one hand with less than 5 digit, extra digit, fewer digits, fused digit, missing digit, bad digit, liquid digit, long body, uncoordinated body, unnatural body, lowres, jpeg artifacts, 3d, cg, text, japanese kanji"
+images = pipe(prompt,negative_prompt=negative_prompt, num_inference_steps=20).images
+images[0].save("girl.png")
 
 ```
 
@@ -174,7 +173,7 @@ Stable Diffusionと同じバイアスが掛かっています。
 - VAEについて
     - Danbooruなどの無断転載サイトを除いた日本の国内法を遵守したデータ: 60万種類 （データ拡張により無限枚作成）
 - U-Netについて
-    - Danbooruなどの無断転載サイトを除いた日本の国内法を遵守したデータ: 100万ペア
+    - Danbooruなどの無断転載サイトを除いた日本の国内法を遵守したデータ: 180万ペ���
 
 **学習プロセス**
 

README_en.md

@@ -3,7 +3,7 @@
 ![eyecatch](eyecatch.jpg)
 
 # Introduction
-Cool Japan Diffusion (for learning) is the latent diffusion model created from Stable Diffsion.
+Cool Japan Diffusion is the latent diffusion model created from Stable Diffsion.
 Cool Japan Diffusion is suitable for genetrating Cool Japan images such as Anime, Manga, and Game.
 
 # Legal and ethical information
@@ -14,9 +14,9 @@ Therefore, we cannot use the model for commercially except for news reporting.
 TBA.
 
 # Usage
-You can try the model by our [Space](https://huggingface.co/spaces/alfredplpl/cool-japan-diffusion-2-1-0).
+You can try the model by our [Space](https://huggingface.co/spaces/aipicasso/cool-japan-diffusion-latest-demo).
 I recommend to use the model by Web UI.
-You can download the model [here](https://huggingface.co/aipicasso/cool-japan-diffusion-2-1-0/resolve/main/v2-1-0.ckpt).
+You can download the model [here](https://huggingface.co/aipicasso/cool-japan-diffusion-2-1-1/resolve/main/v2-1-1.ckpt). Safetensor version is [here](https://huggingface.co/aipicasso/cool-japan-diffusion-2-1-1/resolve/main/v2-1-1.safetensors). Japanese manual is [here](https://alfredplpl.hatenablog.com/entry/2023/01/11/182146).
 
 ## Model Details
 - **Developed by:** Robin Rombach, Patrick Esser, Alfred Increment
@@ -42,7 +42,7 @@ You can download the model [here](https://huggingface.co/aipicasso/cool-japan-di
 - Diffusers
 
 ## Web UI
-Download the model [here]().
+Download the model [here](https://huggingface.co/aipicasso/cool-japan-diffusion-2-1-1/resolve/main/v2-1-1.ckpt).
 Then, install [Web UI](https://github.com/AUTOMATIC1111/stable-diffusion-webui) by AUTIMATIC1111.
 
 ## Diffusers
@@ -56,20 +56,19 @@ pip install --upgrade git+https://github.com/huggingface/diffusers.git transform
 Running the pipeline (if you don't swap the scheduler it will run with the default DDIM, in this example we are swapping it to EulerDiscreteScheduler):
 
 ```python
-from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler
+from diffusers import StableDiffusionPipeline, EulerAncestralDiscreteScheduler
 import torch
 
-model_id = "aipicasso/cool-japan-diffusion-2-1-1-beta"
+model_id = "aipicasso/cool-japan-diffusion-2-1-1"
 
-scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
-pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)
+scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler")
+pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, torch_dtype=torch.float16)#,use_auth_token="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")
 pipe = pipe.to("cuda")
 
-prompt = "anime, a portrait of a girl with black short hair and red eyes, kimono, full color illustration, official art, 4k, detailed"
-negative_prompt="low quality, bad face, bad anatomy, bad hand, lowres, jpeg artifacts, 2d, 3d, cg, text"
-image = pipe(prompt,negative_prompt=negative_prompt).images[0]
-
-image.save("girl.png")
+prompt = "anime, masterpiece, a portrait of a girl, good pupil, 4k, detailed"
+negative_prompt="deformed, blurry, bad anatomy, bad pupil, disfigured, poorly drawn face, mutation, mutated, extra limb, ugly, poorly drawn hands, bad hands, fused fingers, messy drawing, broken legs censor, low quality, mutated hands and fingers, long body, mutation, poorly drawn, bad eyes, ui, error, missing fingers, fused fingers, one hand with more than 5 fingers, one hand with less than 5 fingers, one hand with more than 5 digit, one hand with less than 5 digit, extra digit, fewer digits, fused digit, missing digit, bad digit, liquid digit, long body, uncoordinated body, unnatural body, lowres, jpeg artifacts, 3d, cg, text, japanese kanji"
+images = pipe(prompt,negative_prompt=negative_prompt, num_inference_steps=20).images
+images[0].save("girl.png")
 
 ```
 
@@ -78,4 +77,4 @@ image.save("girl.png")
 - If you have low GPU RAM available, make sure to add a `pipe.enable_attention_slicing()` after sending it to `cuda` for less VRAM usage (to the cost of speed)
 
 
-*This model card was written by: Alfred Increment and is based on the [Stable Diffusion v2](https://huggingface.co/stabilityai/stable-diffusion-2/raw/main/README.md)
+*This model card was written by: Alfred Increment and is based on the [Stable Diffusion v2](https://huggingface.co/stabilityai/stable-diffusion-2/raw/main/README.md)