diff --git a/neurons/miner.py b/neurons/miner.py index 33ac1616..e894bb14 100644 --- a/neurons/miner.py +++ b/neurons/miner.py @@ -141,6 +141,7 @@ async def forward_video( bt.logging.info("Received video challenge!") try: frames_tensor = decode_video_synapse(synapse) + frames_tensor = frames_tensor.to(self.config.neuron.video_detector_device) synapse.prediction = self.video_detector(frames_tensor) except Exception as e: bt.logging.error("Error performing inference")