fix bug
Signed-off-by: zhaohu xing <920232796@qq.com>
This commit is contained in:
parent
52cc83d36b
commit
9c86fb8cac
1 changed files with 5 additions and 1 deletions
|
@ -111,8 +111,12 @@ restricted_opts = {
|
|||
from omegaconf import OmegaConf
|
||||
config = OmegaConf.load(f"{cmd_opts.config}")
|
||||
# XLMR-Large
|
||||
try:
|
||||
text_model_name = config.model.params.cond_stage_config.params.name
|
||||
|
||||
except :
|
||||
text_model_name = "stable_diffusion"
|
||||
|
||||
cmd_opts.disable_extension_access = (cmd_opts.share or cmd_opts.listen or cmd_opts.server_name) and not cmd_opts.enable_insecure_extension_access
|
||||
|
||||
devices.device, devices.device_interrogate, devices.device_gfpgan, devices.device_swinir, devices.device_esrgan, devices.device_scunet, devices.device_codeformer = \
|
||||
|
|
Loading…
Reference in a new issue