From 584addb90eec9769c8d415c0c0df527a77b42ae8 Mon Sep 17 00:00:00 2001 From: Rutger Date: Fri, 8 Dec 2023 15:15:46 +0100 Subject: [PATCH] include model_name in config file --- src/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/main.py b/src/main.py index 0dcea1f3..c24a7246 100644 --- 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