We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a5cedea commit e03eee1Copy full SHA for e03eee1
iotdb-core/ainode/iotdb/ainode/core/inference/inference_request_pool.py
@@ -120,7 +120,9 @@ def _step(self):
120
grouped_requests = list(grouped_requests.values())
121
122
for requests in grouped_requests:
123
- batch_inputs = self._batcher.batch_request(requests).to(self.device)
+ batch_inputs = self._batcher.batch_request(requests).to(
124
+ "cpu"
125
+ ) # The input data should first load to CPU in current version
126
if isinstance(self._inference_pipeline, ForecastPipeline):
127
batch_output = self._inference_pipeline.forecast(
128
batch_inputs,
0 commit comments