Upload scripts/inference.py with huggingface_hub

scripts/inference.py

@@ -34,6 +34,7 @@ def main() -> None:
 
     payload = torch.load(checkpoint, map_location=device)
     ckpt_cfg = payload.get("config", cfg)
+    ckpt_cfg["paths"]["data_dir"] = cfg["paths"]["data_dir"]
     model = build_model(ckpt_cfg).to(device)
     model.load_state_dict(payload["model_state"])
     model.eval()