File size: 281 Bytes
89744f2 |
1 2 3 4 5 6 7 8 9 10 |
# 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")
|