From 7d46a193dccc830c77bc2328bb80bd860891a51f Mon Sep 17 00:00:00 2001
From: mayukhdeb <mayukhmainak2000@gmail.com>
Date: Wed, 3 Apr 2024 23:52:31 -0700
Subject: [PATCH] enable aspect ratio bucketing

---
 training_args.json | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/training_args.json b/training_args.json
index c1059b3..001899a 100644
--- a/training_args.json
+++ b/training_args.json
@@ -32,5 +32,6 @@
     "hard_pivot": false,
     "weight_type": "bf16",
     "dataloader_num_workers": 0,
-    "aspect_ratio_bucketing": false
+    "aspect_ratio_bucketing": true,
+    "do_cache": false
 }
\ No newline at end of file