From 36583855237843ec79a9f63b74ed86a1fb9d68d4 Mon Sep 17 00:00:00 2001 From: renesweet24 Date: Fri, 31 May 2024 11:46:24 -0500 Subject: [PATCH] fix: weights to torch --- bittensor/utils/weight_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bittensor/utils/weight_utils.py b/bittensor/utils/weight_utils.py index 7a3d489ca6..0a46e5ea37 100644 --- a/bittensor/utils/weight_utils.py +++ b/bittensor/utils/weight_utils.py @@ -251,7 +251,7 @@ def process_weights_for_netuid( metagraph = subtensor.metagraph(netuid) # Cast weights to floats. - if not use_torch(): + if use_torch(): if not isinstance(weights, torch.FloatTensor): weights = weights.type(torch.float32) else: