text-generation-inference/backends/gaudi/server/text_generation_server/models
Yuan Wu 3d059f91ab
Gaudi: Use exponential growth to replace BATCH_BUCKET_SIZE (#3131)
* Gaudi: Use exponential growth to replace BATCH_BUCKET_SIZE

Signed-off-by: yuanwu <yuan.wu@intel.com>

* Remove debug modifications

Signed-off-by: yuanwu <yuan.wu@intel.com>

---------

Signed-off-by: yuanwu <yuan.wu@intel.com>
2025-04-03 10:34:53 +02:00
..
custom_modeling Gaudi: Fix llava-next and mllama crash issue (#3127) 2025-03-25 15:08:15 +01:00
__init__.py Gaudi: Sync TGI with the latest changes from the TGI-Gaudi fork (#3117) 2025-03-18 09:45:52 +01:00
bloom.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
causal_lm.py Gaudi: Use exponential growth to replace BATCH_BUCKET_SIZE (#3131) 2025-04-03 10:34:53 +02:00
flash_causal_lm.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
galactica.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
globals.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
idefics_causal_lm.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
mamba.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
mllama_causal_lm.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
model.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
pali_gemma.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
seq2seq_lm.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
starcoder.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
types.py Add Gaudi Backend (#3055) 2025-02-28 12:14:58 +01:00
vlm_causal_lm.py Gaudi: Fix llava-next and mllama crash issue (#3127) 2025-03-25 15:08:15 +01:00