CalvinBackup/llmsecops-research
1
0
Fork 0
mirror of https://github.com/lightbroker/llmsecops-research.git synced 2026-04-30 07:17:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

support HF

Browse Source
This commit is contained in:
Adam Wilson
2025-08-19 11:35:03 -06:00
parent 82e546d5ce
commit ee9355e757
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/text_generation/adapters/foundation_models/meta_llama_foundation_model.py
+7
View File
@@ -1,8 +1,15 @@
import os
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
from huggingface_hub import login
if "HF_TOKEN" in os.environ:
login(token=os.environ["HF_TOKEN"])
else:
raise ValueError("HF_TOKEN environment variable not found")
class MetaLlamaFoundationModel(BaseFoundationModel):