huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-21 23:12:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

More qwen2.

Browse Source
This commit is contained in:
Nicolas Patry 2025-03-12 02:29:31 +01:00
parent 1578cdd9a8
commit 750b48c730
No known key found for this signature in database
GPG Key ID: 4242CEF24CB6DBF9
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
server/text_generation_server/models/custom_modeling/qwen2_vl.py
View File

@ -542,6 +542,7 @@ class Qwen2VLForConditionalGeneration(nn.Module):
max_s=max_s,
true_max_s=max_s,
prefill_cache_indices=prefill_cache_indices,
adapter_data=adapter_data,
)
if lm_head_indices is not None:
hidden_states = hidden_states[lm_head_indices]