huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-08 19:04:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change batch size sort in Neuron config

Browse Source 
Fixing https://github.com/huggingface/text-generation-inference/issues/3299
This commit is contained in:
Jim Burtoft 2025-07-18 22:39:45 -04:00 committed by GitHub
parent 24c2bff659
commit 3c55c053ee
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backends/neuron/server/text_generation_server/tgi_env.py
View File

@ -112,7 +112,7 @@ def neuron_config_to_env(neuron_config):
def sort_neuron_configs(dictionary):
return -dictionary["tp_degree"], -dictionary["batch_size"]
return -dictionary["tp_degree"], dictionary["batch_size"]
def lookup_compatible_cached_model(