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(