xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-13 22:41:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source 
Signed-off-by: Tyler Michael Smith <tlrmchlsmth@gmail.com>
This commit is contained in:
Tyler Michael Smith 2025-08-06 23:53:32 +00:00
parent c41ea52634
commit ddb65dad96
2 changed files with 2 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
requirements/test.txt
View File

@ -178,7 +178,6 @@ einops==0.8.1
# via # via
# -r requirements/test.in # -r requirements/test.in
# encodec # encodec
# mamba-ssm
# terratorch # terratorch
# torchgeo # torchgeo
# vector-quantize-pytorch # vector-quantize-pytorch
@ -418,8 +417,6 @@ lxml==5.3.0
# sacrebleu # sacrebleu
mako==1.3.10 mako==1.3.10
# via alembic # via alembic
mamba-ssm==2.2.5
# via -r requirements/test.in
markdown==3.8.2 markdown==3.8.2
# via mlflow # via mlflow
markdown-it-py==3.0.0 markdown-it-py==3.0.0
@ -476,8 +473,6 @@ networkx==3.2.1
# via # via
# scikit-image # scikit-image
# torch # torch
ninja==1.11.1.3
# via mamba-ssm
nltk==3.9.1 nltk==3.9.1
# via rouge-score # via rouge-score
num2words==0.5.14 num2words==0.5.14
@ -630,7 +625,6 @@ packaging==24.2
# lazy-loader # lazy-loader
# lightning # lightning
# lightning-utilities # lightning-utilities
# mamba-ssm
# matplotlib # matplotlib
# mlflow-skinny # mlflow-skinny
# peft # peft
@ -974,7 +968,6 @@ sentencepiece==0.2.0
setuptools==77.0.3 setuptools==77.0.3
# via # via
# lightning-utilities # lightning-utilities
# mamba-ssm
# pytablewriter # pytablewriter
# torch # torch
# triton # triton
@ -1086,7 +1079,6 @@ torch==2.7.1+cu128
# lightly # lightly
# lightning # lightning
# lm-eval # lm-eval
# mamba-ssm
# mteb # mteb
# open-clip-torch # open-clip-torch
# peft # peft
@ -1153,16 +1145,13 @@ transformers==4.55.0
# -r requirements/test.in # -r requirements/test.in
# genai-perf # genai-perf
# lm-eval # lm-eval
# mamba-ssm
# peft # peft
# sentence-transformers # sentence-transformers
# transformers-stream-generator # transformers-stream-generator
transformers-stream-generator==0.0.5 transformers-stream-generator==0.0.5
# via -r requirements/test.in # via -r requirements/test.in
triton==3.3.1 triton==3.3.1
# via # via torch
# mamba-ssm
# torch
tritonclient==2.51.0 tritonclient==2.51.0
# via # via
# -r requirements/test.in # -r requirements/test.in

1
tests/models/language/generation/test_hybrid.py
View File

@ -25,6 +25,7 @@ SSM_MODELS = [
HYBRID_MODELS = [ HYBRID_MODELS = [
"ai21labs/Jamba-tiny-dev", "ai21labs/Jamba-tiny-dev",
"pfnet/plamo-2-1b",
"Zyphra/Zamba2-1.2B-instruct", "Zyphra/Zamba2-1.2B-instruct",
"hmellor/tiny-random-BambaForCausalLM", "hmellor/tiny-random-BambaForCausalLM",
"ibm-ai-platform/Bamba-9B-v1", "ibm-ai-platform/Bamba-9B-v1",