Jianghai 1622031058 [pipeline] Llama pipeline (#4205) ...

* bloom policy

* llama pipeline forward and tests

* fix the output and attention_mask

* fix name

* bind argument to policy

* Revert "bloom policy"

This reverts commit 8dee68a0a2.

This policy should be revert and copied to feature/bloom

* revert the bloom changes

* cancel unneeded inputs

* gpt

2023-08-15 23:25:14 +08:00

..

model_zoo

[pipeline] Llama pipeline (#4205)

2023-08-15 23:25:14 +08:00

__init__.py

[test] added timm models to test model zoo (#3129)

2023-03-14 14:29:18 +08:00