FEAT: support deepseek-v2 and 2.5 #2292
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
测试环境: NVIDIA GeForce RTX 3090 Ti * 2
![image](https://private-user-images.githubusercontent.com/128140880/367206812-db23c0a2-328a-4eeb-8ec3-d951a5460e4e.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3Mzk3NzU0NTYsIm5iZiI6MTczOTc3NTE1NiwicGF0aCI6Ii8xMjgxNDA4ODAvMzY3MjA2ODEyLWRiMjNjMGEyLTMyOGEtNGVlYi04ZWMzLWQ5NTFhNTQ2MGU0ZS5wbmc_WC1BbXotQWxnb3JpdGhtPUFXUzQtSE1BQy1TSEEyNTYmWC1BbXotQ3JlZGVudGlhbD1BS0lBVkNPRFlMU0E1M1BRSzRaQSUyRjIwMjUwMjE3JTJGdXMtZWFzdC0xJTJGczMlMkZhd3M0X3JlcXVlc3QmWC1BbXotRGF0ZT0yMDI1MDIxN1QwNjUyMzZaJlgtQW16LUV4cGlyZXM9MzAwJlgtQW16LVNpZ25hdHVyZT0zNzYxNTVhOGJhMmEzZGMyNDgxYWVmNGNhMWZiNDBjNTNjZDNjM2Y2MWZhNzljZTg4M2RlNmRjMjE4NmUyMjQxJlgtQW16LVNpZ25lZEhlYWRlcnM9aG9zdCJ9.O5RAB_48bKvXYFyypMNHN8S1-7pQhCQRqrYJ72XaFDU)
![image](https://private-user-images.githubusercontent.com/128140880/367212887-01e9356f-e735-4b49-a01a-21988c835890.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3Mzk3NzU0NTYsIm5iZiI6MTczOTc3NTE1NiwicGF0aCI6Ii8xMjgxNDA4ODAvMzY3MjEyODg3LTAxZTkzNTZmLWU3MzUtNGI0OS1hMDFhLTIxOTg4YzgzNTg5MC5wbmc_WC1BbXotQWxnb3JpdGhtPUFXUzQtSE1BQy1TSEEyNTYmWC1BbXotQ3JlZGVudGlhbD1BS0lBVkNPRFlMU0E1M1BRSzRaQSUyRjIwMjUwMjE3JTJGdXMtZWFzdC0xJTJGczMlMkZhd3M0X3JlcXVlc3QmWC1BbXotRGF0ZT0yMDI1MDIxN1QwNjUyMzZaJlgtQW16LUV4cGlyZXM9MzAwJlgtQW16LVNpZ25hdHVyZT01OWJhMDM5MzdiNmRlNGUwNzUzYzJkMThiNzkyZjAzN2I2ZjhhYWEwOWVhNTgwZGI4OTBjMjc5MGFlNTliNzY2JlgtQW16LVNpZ25lZEhlYWRlcnM9aG9zdCJ9.B1OXXv3bBDHsa79lifBoGJy8M_XtO1f33LWtGLp_W2E)
测试模型:DeepSeek-V2-Lite-Chat
推理引擎:vllm==0.5.5
测试参数:--num-prompt 100 --concurrency 50
推理引擎:transformers==4.39.3
测试参数:--num-prompt 50 --concurrency 5
推理引擎:sglang
模型参数:mem_fraction_static=0.8
测试参数:--num-prompt 100 --concurrency 50