diff --git a/prompting/__init__.py b/prompting/__init__.py
index 643f4f1ac..65aed54ea 100644
--- a/prompting/__init__.py
+++ b/prompting/__init__.py
@@ -16,7 +16,7 @@
 # DEALINGS IN THE SOFTWARE.
 
 # Define the version of the template module.
-__version__ = "2.4.0"
+__version__ = "2.4.2"
 version_split = __version__.split(".")
 __spec_version__ = (
     (10000 * int(version_split[0]))
diff --git a/prompting/base/validator.py b/prompting/base/validator.py
index 893bf2055..93240ad5b 100644
--- a/prompting/base/validator.py
+++ b/prompting/base/validator.py
@@ -239,14 +239,14 @@ def set_weights(self):
         raw_weights = torch.nn.functional.normalize(self.scores, p=1, dim=0)
 
         bt.logging.debug("raw_weights", raw_weights)
-        bt.logging.debug("raw_weight_uids", self.metagraph.uids.to("cpu"))
+        bt.logging.debug("raw_weight_uids", self.metagraph.uids)
         # Process the raw weights to final_weights via subtensor limitations.
         (
             processed_weight_uids,
             processed_weights,
         ) = bt.utils.weight_utils.process_weights_for_netuid(
-            uids=self.metagraph.uids.to("cpu"),
-            weights=raw_weights.to("cpu"),
+            uids=self.metagraph.uids,
+            weights=raw_weights.to("cpu").numpy(),
             netuid=self.config.netuid,
             subtensor=self.subtensor,
             metagraph=self.metagraph,
diff --git a/prompting/utils/config.py b/prompting/utils/config.py
index 4ffa0ddd4..a9382039c 100644
--- a/prompting/utils/config.py
+++ b/prompting/utils/config.py
@@ -307,7 +307,7 @@ def add_validator_args(cls, parser):
         "--neuron.timeout",
         type=float,
         help="The timeout for each forward call in seconds.",
-        default=10,
+        default=17,
     )
 
     parser.add_argument(