fixes torch type issues (#1455)

fixes tensor issues
opentensor · Jul 17, 2023 · abbc90f · abbc90f
1 parent e821bfb
commit abbc90f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/bittensor/tensor.py b/bittensor/tensor.py
@@ -26,8 +26,11 @@
 from typing import Dict, Optional, Tuple, Union, List, Callable
 
 TORCH_DTYPES = {
+    'torch.float16': torch.float16,
     'torch.float32': torch.float32,
     'torch.float64': torch.float64,
+    'torch.uint8': torch.uint8,
+    'torch.int32': torch.int32,
     'torch.int64': torch.int64,
 }