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,
|