Spaces:
Running
on
Zero
Running
on
Zero
Update demo/infer.py
Browse files- 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)
|