# requirements.txt

# PyTorch (CUDA or CPU version). For GPU install, see PyTorch docs for the correct wheel.
torch>=1.12

# MONAI v1.2+ has the 'generative' subpackage with AutoencoderKL, PatchDiscriminator, etc.
monai-weekly
monai-generative

# Common Python libraries
pandas
numpy
nibabel
matplotlib
datasets