Co-authored-by: Satyam Kumar <satyamk@linkedin.com>
This commit is contained in:
co-authored by
Satyam Kumar
parent
c11b34d599
commit
9fc3e8aac7
@@ -203,7 +203,10 @@ class SchedulerOutputProcessorMixin:
|
||||
i
|
||||
].item()
|
||||
else:
|
||||
embeddings = embeddings.tolist()
|
||||
if isinstance(embeddings, torch.Tensor):
|
||||
embeddings = embeddings.tolist()
|
||||
else:
|
||||
embeddings = [tensor.tolist() for tensor in embeddings]
|
||||
|
||||
# Check finish conditions
|
||||
for i, req in enumerate(batch.reqs):
|
||||
|
||||
Reference in New Issue
Block a user