Spaces:
Running
on
Zero
Running
on
Zero
Update demo/infer.py
Browse files- demo/infer.py +1 -1
demo/infer.py
CHANGED
@@ -206,7 +206,7 @@ class LiveCCDemoInfer:
|
|
206 |
return_tensors="pt",
|
207 |
return_attention_mask=False
|
208 |
)
|
209 |
-
inputs.to(
|
210 |
if past_ids is not None:
|
211 |
inputs['input_ids'] = torch.cat([past_ids, inputs.input_ids], dim=1)
|
212 |
outputs = self.model.generate(
|
|
|
206 |
return_tensors="pt",
|
207 |
return_attention_mask=False
|
208 |
)
|
209 |
+
inputs.to(self.model.device)
|
210 |
if past_ids is not None:
|
211 |
inputs['input_ids'] = torch.cat([past_ids, inputs.input_ids], dim=1)
|
212 |
outputs = self.model.generate(
|