kz209 commited on
Commit
ab46b56
·
1 Parent(s): dbe0f74
Files changed (1) hide show
  1. utils/multiple_stream.py +2 -0
utils/multiple_stream.py CHANGED
@@ -29,6 +29,8 @@ def stream_data(content_list, model):
29
  #for i, content in enumerate(content_list):
30
  try:
31
  words = next(model.gen(content_list, streaming=True)) # Wrap content in a list to match expected input type
 
 
32
  outputs = [outputs[i]+f" {words[i]}" for i in range(len(content_list))]
33
  updated = True
34
  except StopIteration:
 
29
  #for i, content in enumerate(content_list):
30
  try:
31
  words = next(model.gen(content_list, streaming=True)) # Wrap content in a list to match expected input type
32
+ print(outputs)
33
+ print(words)
34
  outputs = [outputs[i]+f" {words[i]}" for i in range(len(content_list))]
35
  updated = True
36
  except StopIteration: