Spaces:
Runtime error
Runtime error
Commit ·
107f5e9
1
Parent(s): 0af6b04
Swap gated benjamin-paine SD1.5 for the public stable-diffusion-v1-5/stable-diffusion-v1-5 mirror
Browse files- gradio_app/all_models.py +1 -1
- scripts/sd_model_zoo.py +1 -1
gradio_app/all_models.py
CHANGED
|
@@ -6,7 +6,7 @@ from diffusers import StableDiffusionControlNetImg2ImgPipeline, StableDiffusionP
|
|
| 6 |
class MyModelZoo:
|
| 7 |
_pipe_disney_controlnet_lineart_ipadapter_i2i: StableDiffusionControlNetImg2ImgPipeline = None
|
| 8 |
|
| 9 |
-
base_model = "
|
| 10 |
|
| 11 |
def __init__(self, base_model=None) -> None:
|
| 12 |
if base_model is not None:
|
|
|
|
| 6 |
class MyModelZoo:
|
| 7 |
_pipe_disney_controlnet_lineart_ipadapter_i2i: StableDiffusionControlNetImg2ImgPipeline = None
|
| 8 |
|
| 9 |
+
base_model = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
| 10 |
|
| 11 |
def __init__(self, base_model=None) -> None:
|
| 12 |
if base_model is not None:
|
scripts/sd_model_zoo.py
CHANGED
|
@@ -4,7 +4,7 @@ import torch
|
|
| 4 |
from copy import deepcopy
|
| 5 |
|
| 6 |
ENABLE_CPU_CACHE = False
|
| 7 |
-
DEFAULT_BASE_MODEL = "
|
| 8 |
|
| 9 |
cached_models = {} # cache for models to avoid repeated loading, key is model name
|
| 10 |
def cache_model(func):
|
|
|
|
| 4 |
from copy import deepcopy
|
| 5 |
|
| 6 |
ENABLE_CPU_CACHE = False
|
| 7 |
+
DEFAULT_BASE_MODEL = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
| 8 |
|
| 9 |
cached_models = {} # cache for models to avoid repeated loading, key is model name
|
| 10 |
def cache_model(func):
|