fix

flexflow · Jul 19, 2024 · 55eb7ed · 55eb7ed
1 parent bac88c6
commit 55eb7ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ops/lora_linear.cc b/src/ops/lora_linear.cc
@@ -125,7 +125,7 @@ PEFTModelID *FFModel::add_lora_layer(LoraLinearConfig const peft_config) {
                                             0,
                                             true /*create_grad*/);
         }
-        layers.insert(it + 1, peft_layer);
+        it = layers.insert(it + 1, peft_layer);
         ++it;
         base_layer_to_peft_layer[target_module] = peft_layer;
         peft_layer_to_peft_id[peft_layer] = std::vector<PEFTModelID>();