mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
5cc73f6087
- feature: finetuning your own image models - feature: image prep command. crops to face or other interesting parts of photo - fix: back-compat for hf_hub_download - feature: add prune-ckpt command - feature: allow specification of model config file
104 lines
4.4 KiB
Python
104 lines
4.4 KiB
Python
from dataclasses import dataclass
|
|
|
|
DEFAULT_MODEL = "SD-1.5"
|
|
DEFAULT_SAMPLER = "k_dpmpp_2m"
|
|
|
|
DEFAULT_NEGATIVE_PROMPT = (
|
|
"Ugly, duplication, duplicates, mutilation, deformed, mutilated, mutation, twisted body, disfigured, bad anatomy, "
|
|
"out of frame, extra fingers, mutated hands, "
|
|
"poorly drawn hands, extra limbs, malformed limbs, missing arms, extra arms, missing legs, extra legs, mutated hands, "
|
|
"extra hands, fused fingers, missing fingers, extra fingers, long neck, small head, closed eyes, rolling eyes, "
|
|
"weird eyes, smudged face, blurred face, poorly drawn face, mutation, mutilation, cloned face, strange mouth, "
|
|
"grainy, blurred, blurry, writing, calligraphy, signature, text, watermark, bad art,"
|
|
)
|
|
|
|
SPLITMEM_ENABLED = False
|
|
|
|
|
|
@dataclass
|
|
class ModelConfig:
|
|
short_name: str
|
|
config_path: str
|
|
weights_url: str
|
|
default_image_size: int
|
|
weights_url_full: str = None
|
|
forced_attn_precision: str = "default"
|
|
|
|
|
|
midas_url = "https://github.com/intel-isl/DPT/releases/download/1_0/dpt_hybrid-midas-501f0c75.pt"
|
|
|
|
MODEL_CONFIGS = [
|
|
ModelConfig(
|
|
short_name="SD-1.4",
|
|
config_path="configs/stable-diffusion-v1.yaml",
|
|
weights_url="https://huggingface.co/bstddev/sd-v1-4/resolve/77221977fa8de8ab8f36fac0374c120bd5b53287/sd-v1-4.ckpt",
|
|
default_image_size=512,
|
|
),
|
|
ModelConfig(
|
|
short_name="SD-1.5",
|
|
config_path="configs/stable-diffusion-v1.yaml",
|
|
weights_url="https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/889b629140e71758e1e0006e355c331a5744b4bf/v1-5-pruned-emaonly.ckpt",
|
|
weights_url_full="https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/889b629140e71758e1e0006e355c331a5744b4bf/v1-5-pruned.ckpt",
|
|
default_image_size=512,
|
|
),
|
|
ModelConfig(
|
|
short_name="SD-1.5-inpaint",
|
|
config_path="configs/stable-diffusion-v1-inpaint.yaml",
|
|
weights_url="https://huggingface.co/julienacquaviva/inpainting/resolve/2155ff7fe38b55f4c0d99c2f1ab9b561f8311ca7/sd-v1-5-inpainting.ckpt",
|
|
default_image_size=512,
|
|
),
|
|
ModelConfig(
|
|
short_name="SD-2.0",
|
|
config_path="configs/stable-diffusion-v2-inference.yaml",
|
|
weights_url="https://huggingface.co/stabilityai/stable-diffusion-2-base/resolve/main/512-base-ema.ckpt",
|
|
default_image_size=512,
|
|
),
|
|
ModelConfig(
|
|
short_name="SD-2.0-inpaint",
|
|
config_path="configs/stable-diffusion-v2-inpainting-inference.yaml",
|
|
weights_url="https://huggingface.co/stabilityai/stable-diffusion-2-inpainting/resolve/main/512-inpainting-ema.ckpt",
|
|
default_image_size=512,
|
|
),
|
|
ModelConfig(
|
|
short_name="SD-2.1",
|
|
config_path="configs/stable-diffusion-v2-inference.yaml",
|
|
weights_url="https://huggingface.co/stabilityai/stable-diffusion-2-1-base/resolve/main/v2-1_512-ema-pruned.ckpt",
|
|
default_image_size=512,
|
|
),
|
|
ModelConfig(
|
|
short_name="SD-2.1-inpaint",
|
|
config_path="configs/stable-diffusion-v2-inpainting-inference.yaml",
|
|
weights_url="https://huggingface.co/stabilityai/stable-diffusion-2-inpainting/resolve/main/512-inpainting-ema.ckpt",
|
|
default_image_size=512,
|
|
),
|
|
ModelConfig(
|
|
short_name="SD-2.1-v",
|
|
config_path="configs/stable-diffusion-v2-inference-v.yaml",
|
|
weights_url="https://huggingface.co/stabilityai/stable-diffusion-2-1/resolve/main/v2-1_768-ema-pruned.ckpt",
|
|
default_image_size=768,
|
|
forced_attn_precision="fp32",
|
|
),
|
|
ModelConfig(
|
|
short_name="SD-2.0-v",
|
|
config_path="configs/stable-diffusion-v2-inference-v.yaml",
|
|
weights_url="https://huggingface.co/stabilityai/stable-diffusion-2/resolve/main/768-v-ema.ckpt",
|
|
default_image_size=768,
|
|
),
|
|
ModelConfig(
|
|
short_name="SD-2.0-depth",
|
|
config_path="configs/stable-diffusion-v2-midas-inference.yaml",
|
|
weights_url="https://huggingface.co/stabilityai/stable-diffusion-2-depth/resolve/main/512-depth-ema.ckpt",
|
|
default_image_size=512,
|
|
),
|
|
# ModelConfig(
|
|
# short_name="SD-2.0-upscale",
|
|
# config_path="configs/stable-diffusion-v2-upscaling.yaml",
|
|
# weights_url="https://huggingface.co/stabilityai/stable-diffusion-x4-upscaler/resolve/main/x4-upscaler-ema.ckpt",
|
|
# default_image_size=512,
|
|
# ),
|
|
]
|
|
|
|
MODEL_CONFIG_SHORTCUTS = {m.short_name: m for m in MODEL_CONFIGS}
|
|
|
|
MODEL_SHORT_NAMES = sorted(MODEL_CONFIG_SHORTCUTS.keys())
|