-
Notifications
You must be signed in to change notification settings - Fork 634
[Feat] Support streaming transfer data using ZMQ #3521
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Feat] Support streaming transfer data using ZMQ #3521
Conversation
Wanglongzhi2001 seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account. You have signed the CLA already but the status is still pending? Let us recheck it. |
Thanks for your contribution! |
fastdeploy/output/token_processor.py
Outdated
self.zmq_server = ZmqClient(name=f"get_save_output_rank{self.cfg.local_device_ids[0]}", mode=zmq.PULL) | ||
self.zmq_server.start_server() | ||
self.zmq_server.create_router() | ||
time.sleep(3) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里sleep的目的是为了等待初始化完成还是建联呢 是否有更优雅的方式
|
||
|
||
def post_process_normal( | ||
cls, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这里是不是还有其他硬件的调用也需要去改。以及这里传递zmq_client不行吗 还要把cls传过来
c850e37
to
361c4eb
Compare
6b9ae5b
to
1c5c865
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Support streaming transfer data using ZMQ
Already verified cases:
TODO: