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=truemodel_loaded_successfully=truemlflow_pytorch_pickle_module_import_marker.txtis created duringmlflow.pytorch.load_model()
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support