Spaces:
Running
Running
Ahsen Khaliq
commited on
Commit
•
257cf7d
1
Parent(s):
8bbcb84
Update predict.py
Browse files- predict.py +1 -1
predict.py
CHANGED
@@ -35,7 +35,7 @@ from saicinpainting.utils import register_debug_signal_handlers
|
|
35 |
LOGGER = logging.getLogger(__name__)
|
36 |
|
37 |
|
38 |
-
@hydra.main(config_path='
|
39 |
def main(predict_config: OmegaConf):
|
40 |
try:
|
41 |
register_debug_signal_handlers() # kill -10 <pid> will result in traceback dumped into log
|
|
|
35 |
LOGGER = logging.getLogger(__name__)
|
36 |
|
37 |
|
38 |
+
@hydra.main(config_path='configs/prediction', config_name='default.yaml')
|
39 |
def main(predict_config: OmegaConf):
|
40 |
try:
|
41 |
register_debug_signal_handlers() # kill -10 <pid> will result in traceback dumped into log
|