edit

inference.py

@@ -1,4 +1,6 @@
 from pathlib import Path
+
+import hydra
 import torch
 from omegaconf import DictConfig
 from slider import Beatmap
@@ -8,9 +10,6 @@ from osuT5.inference import Preprocessor, Pipeline, Postprocessor, DiffisionPipe
 from osuT5.tokenizer import Tokenizer
 from osuT5.utils import get_model
 
-config_path = "configs/inference.yaml"
-config = DictConfig.load(config_path)
-
 
 def get_args_from_beatmap(args: DictConfig):
     if args.beatmap_path is None or args.beatmap_path == "":
@@ -49,6 +48,7 @@ def find_model(ckpt_path, args: DictConfig, device):
     return model
 
 
+@hydra.main(config_path="configs", config_name="inference", version_base="1.1")
 def main(args: DictConfig):
     get_args_from_beatmap(args)
 
@@ -114,4 +114,4 @@ def main(args: DictConfig):
 
 
 if __name__ == "__main__":
-    main(config)
+    main()

requirements.txt

@@ -7,4 +7,4 @@ tensorboard
 slider==0.8.1
 torch_tb_profiler
 rosu_pp_py
-omegaconf
+hydra-core