DeepSpeed
9723a879 - Add unit test to check HF low_cpu_mem_usage_flag (#4184)

Commit
2 years ago
Add unit test to check HF low_cpu_mem_usage_flag (#4184) * Add unittest to check huggingface low_cpu_mem_usageflag * change lag to true * Formatting has changes * Indentation fix * Fix chanves * final format fix * Accidently dropped pytestmark from other test * Remove invalid model test config as that was removed. * Whitespace and PR feedback * Format and PR feedback means that we can remove the import we added. * Update tests/unit/inference/test_inference.py * Update tests/unit/inference/test_inference.py --------- Co-authored-by: Lok Chand Koppaka <lokoppak@microsoft.com> Co-authored-by: Michael Wyatt <michaelwyatt@microsoft.com>
Author
Parents
Loading