search

NVIDIA / modulus

Open-source deep-learning framework for building, training, and fine-tuning deep learning models using state-of-the-art Physics-ML methods
https://developer.nvidia.com/modulus
Apache License 2.0
798 stars 174 forks source link

🐛[BUG]: Remove use of pickle module. #437

Closed ktangsali closed 3 months ago

ktangsali commented 3 months ago

Version

0.6.0

On which installation method(s) does this occur?

Pip

Describe the issue

Pickle is currently used here: https://github.com/NVIDIA/modulus/blob/main/modulus/utils/generative/utils.py#L28

Pickle module is not very secure and should be avoided if possible. reference

Minimum reproducible example

No response

Relevant log output

No response

Environment details

No response