瀏覽代碼

Fix trt-fp16 (#3856)

Lin Manhui 7 月之前
父節點
當前提交
b22032a045
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      paddlex/inference/models/common/static_infer.py

+ 1 - 1
paddlex/inference/models/common/static_infer.py

@@ -807,7 +807,7 @@ class HPInfer(StaticInfer):
             cache_dir = self.model_dir / CACHE_DIR / "tensorrt"
             cache_dir.mkdir(parents=True, exist_ok=True)
             ui_option.trt_option.serialize_file = str(cache_dir / "trt_serialized.trt")
-            if backend_config.precision == "FP16":
+            if backend_config.precision == "fp16":
                 ui_option.trt_option.enable_fp16 = True
             if not backend_config.use_dynamic_shapes:
                 raise RuntimeError(