update models

inference.py

@@ -238,11 +238,6 @@ def build_torch_model_from_ckpt(model_name: str, ckpt: dict, device: torch.devic
         model = CNNHead(in_ch=in_dim, c=int(params["channels"]), k=int(params["kernel"]),
                         layers=int(params["layers"]), dropout=dropout)
     elif model_name == "transformer":
-        print(
-            f"[LOAD] {prop_key}/{mode} transformer params keys:",
-            list(params.keys())
-        )
-
         d_model = (
             params.get("d_model")
             or params.get("hidden")