Skip to content

Commit e03eee1

Browse files
committed
Update inference_request_pool.py
1 parent a5cedea commit e03eee1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

iotdb-core/ainode/iotdb/ainode/core/inference/inference_request_pool.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,9 @@ def _step(self):
120120
grouped_requests = list(grouped_requests.values())
121121

122122
for requests in grouped_requests:
123-
batch_inputs = self._batcher.batch_request(requests).to(self.device)
123+
batch_inputs = self._batcher.batch_request(requests).to(
124+
"cpu"
125+
) # The input data should first load to CPU in current version
124126
if isinstance(self._inference_pipeline, ForecastPipeline):
125127
batch_output = self._inference_pipeline.forecast(
126128
batch_inputs,

0 commit comments

Comments
 (0)