Fix bug

PaddlePaddle · Aug 3, 2023 · 7db4ce8 · 7db4ce8
1 parent 06312aa
commit 7db4ce8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/cinn/frontend/paddle_model_to_program.cc b/paddle/cinn/frontend/paddle_model_to_program.cc
@@ -743,7 +743,6 @@ std::unique_ptr<Program> PaddleModelToProgram::operator()(
     const std::string& model_dir, bool is_combined) {
   paddle::cpp::ProgramDesc program_desc;
   if (FLAGS_cinn_infer_model_version < 2.0) {
-    is_combined = false;
     paddle::LoadModelPb(model_dir,
                         "__model__",
                         "",
@@ -753,6 +752,7 @@ std::unique_ptr<Program> PaddleModelToProgram::operator()(
                         false,
                         target_);
   } else {
+    is_combined = true;
     paddle::LoadModelPb(model_dir,
                         ".pdmodel",
                         ".pdiparams",