Update registry.py (#1333)

mosaicml · Jul 3, 2024 · 1641b21 · 1641b21
1 parent 199c3b9
commit 1641b21
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,