You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

MLflow PyTorch pickle_module import RCE

This repository contains a proof-of-concept MLflow PyTorch model artifact.

mlflow.pytorch.load_model() reads data/pickle_module_info.txt, adds the model-controlled code/ directory to sys.path, and imports the named pickle module. The PoC places evil_pickle_module.py in the model's code/ directory and verifies that it executes during model loading.

Run:

python build_and_verify.py

Expected result:

  • marker_created=true
  • model_loaded_successfully=true
  • mlflow_pytorch_pickle_module_import_marker.txt is created during mlflow.pytorch.load_model()
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support