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
|