Commit Graph

6 Commits

Author SHA1 Message Date
Susnato Dhar
ff4e06fd98 changes 2023-11-23 20:14:53 +05:30
Gustavo de Rosa
271c3397ab Update to new model interface. 2023-11-16 17:28:06 +00:00
Gustavo de Rosa
3128bb636a Support for attention_mask in forward pass.
This commit implements the following:

- Cleans up unused arguments and definitions.
- Adds support for `attention_mask`.
- Adds support for cached inference.
2023-09-26 18:17:08 +00:00
Gunasekar
e656142af4 Upload MixFormerSequentialForCausalLM 2023-09-10 06:28:51 +00:00
Gunasekar
ba44a904e2 Upload MixFormerSequentialForCausalLM 2023-09-10 06:15:55 +00:00
Gunasekar
16982066f0 Upload MixFormerSequentialForCausalLM 2023-09-10 05:42:14 +00:00