diff --git a/src/open_clip/transform.py b/src/open_clip/transform.py index e8e20061d..521a203e3 100644 --- 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, )