From 9b0cca6f549356fc369027bd2c712301a91b54a4 Mon Sep 17 00:00:00 2001
From: Ashish Kumar Singh <ashishkmr472@gmail.com>
Date: Thu, 1 Aug 2024 22:15:25 +0000
Subject: [PATCH] feat: also push arguments in config

---
 training_tpu.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/training_tpu.py b/training_tpu.py
index 705d467..a7cbb11 100644
--- a/training_tpu.py
+++ b/training_tpu.py
@@ -983,6 +983,7 @@ def main(argv):
             "temb": (),
             "textcontext": (77, 768)
         },
+        "arguments": args
     }
     
     text_encoders = defaultTextEncodeModel()