Ahsen Khaliq commited on
Commit
257cf7d
1 Parent(s): 8bbcb84

Update predict.py

Browse files
Files changed (1) hide show
  1. 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='./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
 
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