Skip to content

dxa4481/evilModel

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Evil Model

The Llama/llama2 models available for download by facebook are pytorch files, which can be malicious files.

This is because they're pickle files, which can contain arbitrary Python code inside them.

This repo has an example evil ML model, that creates a file on your filesystem.

You can run it with the following:

create the pytorch file:

python evil.py

run the pytorch file:

torchrun --nproc_per_node 1 loader.py

Note the "loader.py" file here is actually non-malicious, the only malicious file is the .pth file, which is the same format llama uses.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages