Update examples/llava/clip.cpp

Co-authored-by: Xuan-Son Nguyen <[email protected]>

Author: alex-jw-brooks

examples/llava/clip.cpp

@@ -763,11 +763,10 @@ static ggml_cgraph * clip_image_build_graph(clip_ctx * ctx, const clip_image_f32
 
         // If this is an embedding feature layer, save the output.
         // NOTE: 0 index here refers to the input to the encoder.
-        for (size_t vl_idx = 0; vl_idx < hparams.vision_feature_layer.size(); vl_idx++) {
-            if (il == ctx->vision_model.hparams.vision_feature_layer[vl_idx]) {
-                embedding_stack.push_back(embeddings);
-                break;
-            }
+        const auto & vision_feature_layer = vision_model.hparams.vision_feature_layer;
+        if (vision_feature_layer.find(il) != vision_feature_layer.end()) {
+            embedding_stack.push_back(embeddings);
+            break;
         }
 
         //const size_t nb_q_w = model.layers[il].q_w->nb[0];