Fix arg not being passed to image_transform from image_transform_v2

mlfoundations · Oct 23, 2023 · 93b2b99 · 93b2b99
1 parent dd73c2e
commit 93b2b99
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/open_clip/transform.py b/src/open_clip/transform.py
@@ -403,4 +403,5 @@ def image_transform_v2(
         interpolation=cfg.interpolation,
         resize_mode=cfg.resize_mode,
         fill_color=cfg.fill_color,
+        aug_cfg=aug_cfg,
     )