Fix gamma error

README.md

@@ -72,7 +72,7 @@ The original transform configuration (shipped in the model config) is:
       "crop": 0.75,
       "dilation": 0.75,
       "erosion": 0.75,
-      "gamma": [-0.5, 0.5],
+      "powerlaw": [-0.5, 0.5],
       "gaussian_blur": 0.75,
       "h_flip": 0.75,
       "hed": [-0.5, 0.5],
@@ -85,7 +85,7 @@ The original transform configuration (shipped in the model config) is:
 }
 ```
 
-* **Continuous transforms** (e.g., `brightness`, `hue`, `hed`, `gamma`, `saturation`) use an **interval** `[min, max]` from which parameters are sampled.
+* **Continuous transforms** (e.g., `brightness`, `hue`, `hed`, `powerlaw`, `saturation`) use an **interval** `[min, max]` from which parameters are sampled.
 * **Discrete/binary transforms** (e.g., `h_flip`, `v_flip`, `dilation`, `erosion`, `rotation`, `gaussian_blur`, `crop`) use a **probability** (e.g., `0.75`) indicating how likely the transform is applied during sampling.
 
 > You can access and modify this at runtime via:
@@ -233,7 +233,7 @@ Then in your offline job/script:
 
 ```python
 from transformers import AutoModel
-cross_transformer = AutoModel.from_pretrained(
+model = AutoModel.from_pretrained(
     "./histaug-conch_v15",   # local path instead of hub ID
     trust_remote_code=True,
     local_files_only=True,  # uses local files only

config.json

@@ -28,7 +28,7 @@
       "crop": 0.75,
       "dilation": 0.75,
       "erosion": 0.75,
-      "gamma": [
+      "powerlaw": [
         -0.5,
         0.5
       ],

histaug_model.py

@@ -317,7 +317,7 @@ class HistaugModel(nn.Module):
                 "saturation",
                 "hed",
                 "hue",
-                "gamma",
+                "powerlaw",
             ]:
                 # Continuous transformations
                 transform_embeddings[aug_name] = nn.Sequential(
@@ -355,7 +355,7 @@ class HistaugModel(nn.Module):
                 "contrast",
                 "saturation",
                 "hue",
-                "gamma",
+                "powerlaw",
                 "hed",
             ]:
                 z_transform = self.transform_embeddings[aug_name](
@@ -414,7 +414,7 @@ class HistaugModel(nn.Module):
             "contrast",
             "saturation",
             "hue",
-            "gamma",
+            "powerlaw",
             "hed",
         ]
 
@@ -509,7 +509,7 @@ class HistaugModel(nn.Module):
                 "contrast",
                 "saturation",
                 "hue",
-                "gamma",
+                "powerlaw",
                 "hed",
             ):
                 lower_bound, upper_bound = map(float, config)
@@ -566,4 +566,4 @@ class HistaugModel(nn.Module):
         x = x.view(x.shape[0], 1, -1)
         x = self.head(x)
         x = x[:, 0, :]
-        return x
+        return x

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7b77eb00216ae97b0ea01281d2b4cb994e6614b7a62f4831cfa7d18c6c1a5e89
-size 13111600
+oid sha256:f09dd64a7c30dd3fae745c0c645cb3efd2ea69386618745d6d0bd2f4375025e8
+size 13111616