sky-admin commited on
Commit
3c18c09
·
1 Parent(s): 5b824cf
Files changed (1) hide show
  1. predict.py +2 -2
predict.py CHANGED
@@ -56,11 +56,10 @@ def initialize():
56
  gfpgan.setup_model(cmd_opts.gfpgan_models_path)
57
 
58
  modelloader.list_builtin_upscalers()
 
59
  modelloader.load_upscalers()
60
-
61
  modules.sd_vae.refresh_vae_list()
62
 
63
-
64
  try:
65
  modules.sd_models.load_model()
66
  except Exception as e:
@@ -103,6 +102,7 @@ class Predictor(BasePredictor):
103
  seed: int = Input(description="seed", default=-1),
104
  ) -> Path:
105
  """Run a single prediction on the model"""
 
106
  args = {
107
  "do_not_save_samples": True,
108
  "do_not_save_grid": True,
 
56
  gfpgan.setup_model(cmd_opts.gfpgan_models_path)
57
 
58
  modelloader.list_builtin_upscalers()
59
+ modules.scripts.load_scripts()
60
  modelloader.load_upscalers()
 
61
  modules.sd_vae.refresh_vae_list()
62
 
 
63
  try:
64
  modules.sd_models.load_model()
65
  except Exception as e:
 
102
  seed: int = Input(description="seed", default=-1),
103
  ) -> Path:
104
  """Run a single prediction on the model"""
105
+ modules.script_callbacks.before_ui_callback()
106
  args = {
107
  "do_not_save_samples": True,
108
  "do_not_save_grid": True,