From b311a0a55ee7a3a1d9fa9e7c283847aa4634f149 Mon Sep 17 00:00:00 2001 From: Victor Umesiobi Date: Tue, 31 Dec 2024 20:16:11 +0000 Subject: [PATCH] clean ups --- src/latent_gemma/__init__.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/latent_gemma/__init__.py b/src/latent_gemma/__init__.py index b29343f..fb345d4 100644 --- a/src/latent_gemma/__init__.py +++ b/src/latent_gemma/__init__.py @@ -1,5 +1,4 @@ -from .main import LatentReasoningGemmaForCausalLM, LatentReasoningCustomModelForCausalLM, LatentReasoningForCausalLM, LatentReasoningGPT2 - +from .main import LatentReasoningGemmaForCausalLM # Version info __version__ = "0.1.0"