From 6505205b1342ef3ff1a6675f911ba5050cdd44a4 Mon Sep 17 00:00:00 2001 From: Adam Wilson Date: Tue, 19 Aug 2025 11:02:21 -0600 Subject: [PATCH] enum fix --- .../foundation_models/apple_openelm_foundation_model.py | 2 +- .../foundation_models/meta_llama_foundation_model.py | 9 +++------ .../foundation_models/microsoft_phi3_foundation_model.py | 1 - 3 files changed, 4 insertions(+), 8 deletions(-) diff --git a/src/text_generation/adapters/foundation_models/apple_openelm_foundation_model.py b/src/text_generation/adapters/foundation_models/apple_openelm_foundation_model.py index c51ad636b..dbfee306f 100644 --- a/src/text_generation/adapters/foundation_models/apple_openelm_foundation_model.py +++ b/src/text_generation/adapters/foundation_models/apple_openelm_foundation_model.py @@ -8,7 +8,7 @@ from src.text_generation.common.model_id import ModelId class AppleOpenELMFoundationModel(BaseFoundationModel): """apple/OpenELM-3B-Instruct implementation""" - MODEL_ID = ModelId.APPLE_OPENELM_3B_INSTRUCT.value + MODEL_ID = ModelId.APPLE_OPENELM_3B_INSTRUCT def __init__(self, config: AppleOpenELMConfig = AppleOpenELMConfig()): self.config = config diff --git a/src/text_generation/adapters/foundation_models/meta_llama_foundation_model.py b/src/text_generation/adapters/foundation_models/meta_llama_foundation_model.py index 199842c4e..beedc12ae 100644 --- a/src/text_generation/adapters/foundation_models/meta_llama_foundation_model.py +++ b/src/text_generation/adapters/foundation_models/meta_llama_foundation_model.py @@ -1,17 +1,14 @@ -from src.text_generation.adapters.foundation_models.config.meta_llama_config import MetaLlamaConfig -from src.text_generation.adapters.foundation_models.base.base_foundation_model import BaseFoundationModel - - from langchain_huggingface import HuggingFacePipeline from transformers import AutoModelForCausalLM, AutoTokenizer - +from src.text_generation.adapters.foundation_models.config.meta_llama_config import MetaLlamaConfig +from src.text_generation.adapters.foundation_models.base.base_foundation_model import BaseFoundationModel from src.text_generation.common.model_id import ModelId class MetaLlamaFoundationModel(BaseFoundationModel): """meta-llama/Llama-3.2-3B-Instruct implementation""" - MODEL_ID = ModelId.META_LLAMA_3_2_3B_INSTRUCT.value + MODEL_ID = ModelId.META_LLAMA_3_2_3B_INSTRUCT def __init__(self, config: MetaLlamaConfig = MetaLlamaConfig()): self.config = config diff --git a/src/text_generation/adapters/foundation_models/microsoft_phi3_foundation_model.py b/src/text_generation/adapters/foundation_models/microsoft_phi3_foundation_model.py index 68e8726d8..25451a6fb 100644 --- a/src/text_generation/adapters/foundation_models/microsoft_phi3_foundation_model.py +++ b/src/text_generation/adapters/foundation_models/microsoft_phi3_foundation_model.py @@ -1,5 +1,4 @@ from langchain_huggingface import HuggingFacePipeline -from optimum.onnxruntime import ORTModelForCausalLM from transformers import AutoModelForCausalLM, AutoTokenizer from src.text_generation.adapters.foundation_models.config.microsoft_phi3mini4k_config import MicrosoftPhi3Mini4KConfig from src.text_generation.adapters.foundation_models.base.base_foundation_model import BaseFoundationModel