diff --git a/app/sana_pipeline.py b/app/sana_pipeline.py
index cfbec8e..43417ae 100644
--- a/app/sana_pipeline.py
+++ b/app/sana_pipeline.py
@@ -77,7 +77,8 @@ def __init__(
         config: Optional[str] = "configs/sana_config/1024ms/Sana_1600M_img1024.yaml",
     ):
         super().__init__()
-        config = pyrallis.parse(config_class=SanaInference, config_path=config)
+        with open(config) as cf:
+            config = pyrallis.load(SanaInference, f)
         self.args = self.config = config
 
         # set some hyper-parameters