Use `device_map="auto"` in single file tests to support large models on limited GPU memory (#13816)
* fix flux tests OOM on 24G GPU
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
* revert wrong change
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
---------
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>