vllm/worker/pooling_model_runner.py · 3de2b1eafb12e420c563cb7153d4d2f0e8451ca9 · OpenDAS / vllm_cscc

[Model] Add JambaForSequenceClassification model (#10860) · 6c7f8815

Yehoshua Cohen authored Dec 19, 2024


Signed-off-by: Yehoshua Cohen <yehoshuaco@ai21.com>
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
Co-authored-by: Yehoshua Cohen <yehoshuaco@ai21.com>
Co-authored-by: DarkLight1337 <tlleungac@connect.ust.hk>

6c7f8815

pooling_model_runner.py 8.45 KB

Replace pooling_model_runner.py