add MPS support to base_trainer.py#151

francescomalandrino wants to merge 1 commit into

clementchadebec:mainclementchadebec/benchmark_VAE:mainfrom

francescomalandrino:rudimentary-mpsfrancescomalandrino/benchmark_VAE:rudimentary-mpsCopy head branch name to clipboard