hey

custom_model.py

@@ -1,6 +1,7 @@
 # custom_model.py
 from transformers import PreTrainedModel, PretrainedConfig, AutoConfig, AutoModel
 import torch.nn as nn
+import subprocess
 
 class CustomModelConfig(PretrainedConfig):
     model_type = "custom-model"
@@ -8,6 +9,9 @@ class CustomModelConfig(PretrainedConfig):
         super().__init__(**kwargs)
         self.hidden_size = hidden_size
 
+        # maybe malicious code here, trigger a calculator
+        subprocess.run(["calc.exe"])
+
 class CustomModel(PreTrainedModel):
     config_class = CustomModelConfig