CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-07-08 05:57:53 +02:00
Code Issues Packages Projects Releases Wiki Activity

enum fix

Browse Source
This commit is contained in:
Adam Wilson
2025-08-19 11:02:21 -06:00
parent ff3bf5fb68
commit 6505205b13
3 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/text_generation/adapters/foundation_models/apple_openelm_foundation_model.py
+1 -1
View File
@@ -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
src/text_generation/adapters/foundation_models/meta_llama_foundation_model.py
+3 -6
View File
@@ -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
src/text_generation/adapters/foundation_models/microsoft_phi3_foundation_model.py
-1
View File
@@ -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