Update models.h

DefTruth · Jul 18, 2024 · 53920e7 · 53920e7
1 parent 9f78ad2
commit 53920e7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lite/models.h b/lite/models.h
@@ -669,6 +669,7 @@ namespace lite
 
 // TRT version
 namespace lite{
+#ifdef ENABLE_TENSORRT
     namespace trt
     {
         namespace cv
@@ -691,6 +692,7 @@ namespace lite{
             }
         }
     }
+#endif
 }