fix: improve prompt_split_image with ref to original impl

huggingface · Jan 7, 2025 · d397748 · d397748
1 parent df504e9
commit d397748
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/server/text_generation_server/models/vlm_causal_lm.py b/server/text_generation_server/models/vlm_causal_lm.py
@@ -29,6 +29,7 @@
 IDEFICS3_GLOBAL_IMG_TOKEN = "<global-img>"
 
 
+# copied from: https://github.com/huggingface/transformers/blob/02ed609285c2448b3b54c31e362f2c389fa952ab/src/transformers/models/idefics3/processing_idefics3.py#L44-L60
 def get_image_prompt_string(
     *,
     image_seq_len: int,