phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix meta-reference GPU implementation for inference

Browse source
This commit is contained in:
Ashwin Bharambe 2025-01-22 18:31:59 -08:00
parent f4b0f2af8b
commit 23f1980f9c
2 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/inline/inference/meta_reference/parallel_utils.py
View file

@ -357,8 +357,8 @@ class ModelParallelProcessGroup:
assert not self.running, "inference already running"
self.running = True
self.request_socket.send(encode_msg(TaskRequest(task=req)))
try:
self.request_socket.send(encode_msg(TaskRequest(task=req)))
while True:
obj_json = self.request_socket.recv()
obj = parse_message(obj_json)