Update relik/inference/annotator.py

relik/inference/annotator.py

@@ -7,7 +7,6 @@ from typing import Any, Dict, List, Optional, Union
 import hydra
 import torch
 from omegaconf import DictConfig, OmegaConf
-from pprintpp import pformat
 
 from relik.inference.data.splitters.blank_sentence_splitter import BlankSentenceSplitter
 from relik.common.log import get_logger
@@ -621,7 +620,6 @@ class Relik:
         config = OmegaConf.merge(config, OmegaConf.create(kwargs))
         # do we want to print the config? I like it
         logger.info(f"Loading Relik from {model_name_or_dir}")
-        logger.info(pformat(OmegaConf.to_container(config)))
 
         # load relik from config
         relik = hydra.utils.instantiate(config, _recursive_=False, *args)
@@ -823,8 +821,7 @@ class Relik:
         config_file_name = config_file_name or CONFIG_NAME
 
         logger.info(f"Saving relik config to {output_dir / config_file_name}")
-        # pretty print the config
-        logger.info(pformat(config))
+
         OmegaConf.save(config, output_dir / config_file_name)
 
         if push_to_hub: