refactor: removes unused model configs

pull/396/head
jaydrennan 7 months ago committed by Bryce Drennan
parent 76df1210f4
commit e91a041a78

@ -31,14 +31,6 @@ class ModelConfig:
midas_url = "https://github.com/intel-isl/DPT/releases/download/1_0/dpt_hybrid-midas-501f0c75.pt"
MODEL_CONFIGS = [
# ModelConfig(
# description="Stable Diffusion 1.4",
# 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,
# alias="sd14",
# ),
ModelConfig(
description="Stable Diffusion 1.5",
short_name="SD-1.5",
@ -57,63 +49,6 @@ MODEL_CONFIGS = [
alias="sd15in",
),
# ModelConfig(
# description="Stable Diffusion 2.0 - bad at making people",
# 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,
# alias="sd20",
# ),
# ModelConfig(
# description="Stable Diffusion 2.0 - Inpainting",
# 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,
# alias="sd20in",
# ),
# ModelConfig(
# description="Stable Diffusion 2.0 v - 768x768 - bad at making people",
# 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,
# alias="sd20v",
# ),
# ModelConfig(
# description="Stable Diffusion 2.0 - Depth",
# 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,
# alias="sd20dep",
# ),
# ModelConfig(
# description="Stable Diffusion 2.1",
# 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,
# alias="sd21",
# ),
# ModelConfig(
# description="Stable Diffusion 2.1 - Inpainting",
# 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,
# alias="sd21in",
# ),
# ModelConfig(
# description="Stable Diffusion 2.1 v - 768x768",
# 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",
# alias="sd21v",
# ),
# ModelConfig(
# description="Instruct Pix2Pix - Photo Editing",
# short_name="instruct-pix2pix",
# config_path="configs/instruct-pix2pix.yaml",

Loading…
Cancel
Save