instantiate image before augmenting

BitMind-AI · Oct 30, 2024 · 5ee24ed · 5ee24ed
1 parent 7c108bc
commit 5ee24ed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bitmind/validator/forward.py b/bitmind/validator/forward.py
@@ -126,8 +126,9 @@ async def forward(self):
             bt.logging.error(f'unsupported neuron.prompt_type: {self.config.neuron.prompt_type}')
             raise NotImplementedError
 
+    image = sample['image']
     if np.random.rand() > 0.25:
-        image = random_aug_transforms(sample['image'])
+        image = random_aug_transforms(image)
         data_aug_params = random_aug_transforms.params
     else:
         data_aug_params = {}