feat: also push arguments in config

AshishKumar4 · Aug 1, 2024 · 9b0cca6 · 9b0cca6
1 parent b37d853
commit 9b0cca6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/training_tpu.py b/training_tpu.py
@@ -983,6 +983,7 @@ def main(argv):
             "temb": (),
             "textcontext": (77, 768)
         },
+        "arguments": args
     }
 
     text_encoders = defaultTextEncodeModel()