sky-admin
commited on
Commit
·
3c18c09
1
Parent(s):
5b824cf
try fix
Browse files- 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,
|