Update README.md
Browse files
README.md
CHANGED
|
@@ -12,7 +12,7 @@ tags:
|
|
| 12 |
---
|
| 13 |
# ControlNet for Manga Colorization
|
| 14 |
|
| 15 |
-
**Model Name:** `SubMaroon/
|
| 16 |
**Base model:** `John6666/nsfw-anime-xl-v1-sdxl`
|
| 17 |
**Task:** Conditional image generation — Colorization
|
| 18 |
**Conditioning:** Grayscale manga panel (lineart or filled)
|
|
@@ -52,7 +52,7 @@ from diffusers.utils import load_image
|
|
| 52 |
import torch
|
| 53 |
|
| 54 |
# Load ControlNet
|
| 55 |
-
controlnet = ControlNetModel.from_pretrained("SubMaroon/
|
| 56 |
|
| 57 |
# Load base pipeline
|
| 58 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
|
|
|
| 12 |
---
|
| 13 |
# ControlNet for Manga Colorization
|
| 14 |
|
| 15 |
+
**Model Name:** `SubMaroon/ControlNet-manga-recolor`
|
| 16 |
**Base model:** `John6666/nsfw-anime-xl-v1-sdxl`
|
| 17 |
**Task:** Conditional image generation — Colorization
|
| 18 |
**Conditioning:** Grayscale manga panel (lineart or filled)
|
|
|
|
| 52 |
import torch
|
| 53 |
|
| 54 |
# Load ControlNet
|
| 55 |
+
controlnet = ControlNetModel.from_pretrained("SubMaroon/ControlNet-manga-recolor", torch_dtype=torch.float16)
|
| 56 |
|
| 57 |
# Load base pipeline
|
| 58 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|