whoopsies

ZeroCool940711 · Jun 7, 2023 · 729cbad · 729cbad
1 parent 89c35d1
commit 729cbad
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/train_muse_maskgit.py b/train_muse_maskgit.py
@@ -425,7 +425,7 @@ def main():
         print("Using config file and ignoring CLI args")
 
         if args.generate_config:
-            conf = OmegaConf.structured(Arguments)
+            conf = OmegaConf.structured(args)
 
             # dumps to file:
             with open(args.config_path, "w") as f:

diff --git a/train_muse_vae.py b/train_muse_vae.py
@@ -315,7 +315,7 @@ def main():
         print("Using config file and ignoring CLI args")
 
         if args.generate_config:
-            conf = OmegaConf.structured(Arguments)
+            conf = OmegaConf.structured(args)
 
             # dumps to file:
             with open(args.config_path, "w") as f: