diff --git a/代码/code/infer.py b/代码/code/infer.py index 617d1fa..d2109c5 100644 --- a/代码/code/infer.py +++ b/代码/code/infer.py @@ -511,11 +511,6 @@ def load_model(ckpt_path, device='cuda:0'): model.to(dev) model.eval() - - # === torch.compile(default):算子融合,不用 CUDA Graph,兼容动态 batch 形状 === - model = torch.compile(model, mode="default") - print("[INFO] torch.compile applied (mode=default)") - print(f"[INFO] Model ready. Device: {dev}") return model, dev