include model_name in config file

knaw-huc · Dec 8, 2023 · 584addb · 584addb
1 parent 8e4b0a4
commit 584addb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main.py b/src/main.py
@@ -630,7 +630,8 @@ def get_config(args, model):
         'model': model_layers,
         'notes': ' ',
         'uuid': str(uuid.uuid4()),
-        'url-code': 'https://github.com/knaw-huc/loghi'
+        'url-code': 'https://github.com/knaw-huc/loghi',
+        'model_name': args.model_name
     }
     return config