mem-run-estimator-fork To run: python driver.py --test --preset_config --config_idx 0 --model_name stable_diffusion python driver.py --test --preset_config --config_idx 0 --model_name flux