Fix SD obj not defined if fail to get models from pretrained (#1222)

apps/stable_diffusion/web/index.py

@@ -66,7 +66,7 @@ from apps.stable_diffusion.web.ui import (
 )
 
 # init global sd pipeline and config
-global_obj.init()
+global_obj._init()
 
 
 def register_button_click(button, selectedid, inputs, outputs):

apps/stable_diffusion/web/utils/global_obj.py

@@ -8,49 +8,50 @@ Also we could avoid memory leak when switching models by clearing the cache.
 """
 
 
-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
-
-
-def get_sd_obj():
-    return sd_obj
-
-
-def get_cfg_obj():
-    return config_obj
+    global _sd_obj
+    _sd_obj.scheduler = value
 
 
 def set_sd_status(value):
-    global sd_obj
-    sd_obj.status = value
+    global _sd_obj
+    _sd_obj.status = value
+
+
+def set_cfg_obj(value):
+    global _config_obj
+    _config_obj = 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