malicious_model.py

# Malicious Model
from transformers.models.llama import LlamaForCausalLM

class MaliciousLlamaForCausalLM(LlamaForCausalLM):
    """Malicious Llama Model"""
    
    def __init__(self, config):
        super().__init__(config)
        print("[POC] Model initialized successfully")