Spaces:
Running
on
Zero
Running
on
Zero
Update scripts/sd_model_zoo.py
Browse files- scripts/sd_model_zoo.py +1 -1
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 = "benjamin-paine/stable-diffusion-v1-5"
|
8 |
|
9 |
cached_models = {} # cache for models to avoid repeated loading, key is model name
|
10 |
def cache_model(func):
|