Skip to content

Commit

Permalink
Fix SD obj not defined if fail to get models from pretrained (nod-ai#…
Browse files Browse the repository at this point in the history
  • Loading branch information
jinchen62 authored Mar 21, 2023
1 parent 8b8cc7f commit 7dbbb17
Show file tree
Hide file tree
Showing 2 changed files with 29 additions and 28 deletions.
2 changes: 1 addition & 1 deletion apps/stable_diffusion/web/index.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,7 +66,7 @@ def resource_path(relative_path):
)

# init global sd pipeline and config
global_obj.init()
global_obj._init()


def register_button_click(button, selectedid, inputs, outputs):
Expand Down
55 changes: 28 additions & 27 deletions apps/stable_diffusion/web/utils/global_obj.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,49 +8,50 @@
"""


def init():
global sd_obj
global config_obj
sd_obj = None
config_obj = None
def _init():
global _sd_obj
global _config_obj
_sd_obj = None
_config_obj = None


def set_sd_obj(value):
global sd_obj
sd_obj = value


def set_cfg_obj(value):
global config_obj
config_obj = value
global _sd_obj
_sd_obj = value


def set_schedulers(value):
global sd_obj
sd_obj.scheduler = value
global _sd_obj
_sd_obj.scheduler = value


def get_sd_obj():
return sd_obj
def set_sd_status(value):
global _sd_obj
_sd_obj.status = value


def get_cfg_obj():
return config_obj
def set_cfg_obj(value):
global _config_obj
_config_obj = value


def set_sd_status(value):
global sd_obj
sd_obj.status = value
def get_sd_obj():
return _sd_obj


def get_sd_status():
global sd_obj
return sd_obj.status
return _sd_obj.status


def get_cfg_obj():
return _config_obj


def clear_cache():
global sd_obj
global config_obj
del sd_obj
del config_obj
global _sd_obj
global _config_obj
del _sd_obj
del _config_obj
gc.collect()
_sd_obj = None
_config_obj = None

0 comments on commit 7dbbb17

Please sign in to comment.