Wuvin commited on
Commit
b948e08
1 Parent(s): e4f6021

Update scripts/sd_model_zoo.py

Browse files
Files changed (1) hide show
  1. 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 = "runwayml/stable-diffusion-v1-5"
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):