huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-05-21 17:52:09 +00:00
Code Issues Packages Projects Releases Wiki Activity
533eee50dc
text-generation-inference/backends/gaudi/server/text_generation_server/layers/attention
History
Wang, Yi 533eee50dc
forward and tokenize chooser use the same shape (#3196) 
* forward and tokenize chooser use the same shape
concate or filter happened to cpu tensor to avoid dynamic shape in hpu

Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>

* use hpu set seed

Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>

---------

Signed-off-by: Wang, Yi A <yi.a.wang@intel.com>
2025-05-06 10:49:32 +02:00
..
__init__.py forward and tokenize chooser use the same shape (#3196) 2025-05-06 10:49:32 +02:00
common.py forward and tokenize chooser use the same shape (#3196) 2025-05-06 10:49:32 +02:00
hpu.py Warmup gaudi backend (#3172) 2025-04-24 09:57:08 +02:00
kv_cache.py Gaudi: clean cuda/rocm code in hpu backend, enable flat_hpu (#3113) 2025-04-14 15:58:13 +02:00