From 3b151454383d1fa5a3f7b6f70ee4de1303c7e75f Mon Sep 17 00:00:00 2001 From: Aditi Date: Mon, 24 Feb 2025 20:25:48 -0800 Subject: [PATCH] added olmo to init --- pyvene/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyvene/__init__.py b/pyvene/__init__.py index 2527fa4a..8e2934fd 100644 --- a/pyvene/__init__.py +++ b/pyvene/__init__.py @@ -52,4 +52,4 @@ from .models.llama.modelings_intervenable_llama import create_llama from .models.mlp.modelings_intervenable_mlp import create_mlp_classifier from .models.backpack_gpt2.modelings_intervenable_backpack_gpt2 import create_backpack_gpt2 - +from .models.olmo.modelings_intervenable_olmo import create_olmo