chenjoya commited on
Commit
9abe525
·
verified ·
1 Parent(s): 997bb7e

Update demo/infer.py

Browse files
Files changed (1) hide show
  1. demo/infer.py +1 -0
demo/infer.py CHANGED
@@ -146,6 +146,7 @@ class LiveCCDemoInfer:
146
  return_tensors="pt",
147
  return_attention_mask=False
148
  )
 
149
  inputs.to(self.model.device)
150
  if past_ids is not None:
151
  inputs['input_ids'] = torch.cat([past_ids, inputs.input_ids], dim=1)
 
146
  return_tensors="pt",
147
  return_attention_mask=False
148
  )
149
+ print(texts)
150
  inputs.to(self.model.device)
151
  if past_ids is not None:
152
  inputs['input_ids'] = torch.cat([past_ids, inputs.input_ids], dim=1)