[fix] input_augmentation_bypass for remote inference

Author: jianwensong

compressai_vision/datasets/image.py

@@ -263,9 +263,13 @@ def __init__(self, root, dataset_name, imgs_folder, **kwargs):
 
         if self.input_agumentation_bypass:
             emptyAugList = AugmentationList([])
-            self._org_mapper_func.augmentations = emptyAugList
             mapper.augmentations = emptyAugList
 
+            if hasattr(self._org_mapper_func, "_obj"):
+                self._org_mapper_func._obj.augmentations = emptyAugList
+            else:
+                self._org_mapper_func.augmentations = emptyAugList
+
         self.mapDataset = MapDataset(_dataset, mapper)
 
         metaData = MetadataCatalog.get(dataset_name)