From 1641b2176cd15a04e109a5c62b3da80e3df08e91 Mon Sep 17 00:00:00 2001 From: Mihir Patel Date: Tue, 2 Jul 2024 17:53:53 -0700 Subject: [PATCH] Update registry.py (#1333) --- llmfoundry/registry.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llmfoundry/registry.py b/llmfoundry/registry.py index 5924070497..50481211ac 100644 --- a/llmfoundry/registry.py +++ b/llmfoundry/registry.py @@ -83,7 +83,8 @@ """ ) callbacks_with_config = create_registry( - 'llm_foundry.callbacks_with_config', + 'llmfoundry', + 'callbacks_with_config', generic_type=Type[CallbackWithConfig], entry_points=True, description=_callbacks_with_config_description,