server/text_generation_server/models/causal_lm.py · b4cf832c40f1fc17401968198b739bdf8351d9f1 · OpenDAS / text-generation-inference · GitLab

Find file Blame History Permalink

fix(server): fix reshaping of bloom past_key_values in concatenate() (#252) · b4cf832c
Nick Hill authored Apr 27, 2023
```
Introduced in #214 

Fixes #249
```
b4cf832c

causal_lm.py 23.2 KB