Fix arg not being passed to image_transform from image_transform_v2

mlfoundations · Oct 23, 2023 · e850caa · e850caa
1 parent 6d158ea
commit e850caa
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,
     )