Update modeling_jais.py for compatibility with transformers
#20
by
sylwia-kuros
- opened
This change fixes the following issue:
File "/.cache/huggingface/modules/transformers_modules/modeling_jais.py", line 820, in forward
past_length = past_key_values[0][0].size(-2)
AttributeError: 'NoneType' object has no attribute 'size