From 67cade84716b997a0e3c3f6058625cdeb89f0092 Mon Sep 17 00:00:00 2001 From: Douglas Jacobsen Date: Sat, 16 Nov 2024 13:36:52 -0700 Subject: [PATCH] py-nemo: Add generated nemo.yaml to archive --- var/ramble/repos/builtin/applications/py-nemo/application.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/var/ramble/repos/builtin/applications/py-nemo/application.py b/var/ramble/repos/builtin/applications/py-nemo/application.py index 12ba8ea70..de28c2dc6 100644 --- a/var/ramble/repos/builtin/applications/py-nemo/application.py +++ b/var/ramble/repos/builtin/applications/py-nemo/application.py @@ -27,6 +27,8 @@ class PyNemo(ExecutableApplication): tags("ml-framework", "machine-learning") + archive_pattern("{experiment_run_dir}/{nemo_generated_config_name}") + executable( "setup_transformer_cache", 'bash -c "python3 -c \'from transformers import AutoTokenizer; AutoTokenizer.from_pretrained(\\"gpt2\\")\'"',