From e850caa6aed6f57de5249cf2ce030dad2bfe65a3 Mon Sep 17 00:00:00 2001
From: Santiago Castro <sacastro@umich.edu>
Date: Mon, 23 Oct 2023 15:24:14 -0400
Subject: [PATCH] Fix arg not being passed to `image_transform` from
 `image_transform_v2`

---
 src/open_clip/transform.py | 1 +
 1 file changed, 1 insertion(+)

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,
     )