issue/189: add inference server support to InfiniLM #190
+1,276
−0
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
PanZezhong1725
requested changes
Jan 15, 2026
| def start(self): | ||
| app = self._create_app() | ||
| logger.info("Starting API Server...") | ||
| uvicorn.run(app, host="0.0.0.0", port=8000) |
Collaborator
There was a problem hiding this comment.
把port做一个脚本参数，默认8000
|
|
||
| def _create_app(self): | ||
| @asynccontextmanager | ||
| async def lifespan(app: FastAPI): |
Collaborator
There was a problem hiding this comment.
请模仿vllm.LLM或sglang.Engine对这部分逻辑进行封装，server的脚本尽量只传递request，不要出现这种业务逻辑相关的代码
这个封装应该提供两种用法：
- 面向网络服务，异步流式输出，即server脚本的用法
- 面向单机使用，提供batch generate的方法，用户可同时传入多个请求，返回所有结果。目前需要这个的使用场景就是让ceval测试可以批量化进行
请把当前的core/目录按照封装重新命名，封装的代码也放到里面
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
9G7B 单卡 max_batch_size=32：
9G7B 4卡 max_batch_size=32：
