Deepspeed Mistral, Memory profiling support and empirical quality checks

premAI-io / benchmarks

🕹️ Performance Comparison of MLOps Engines, Frameworks, and Languages on Mainstream AI Models.

MIT License

117 stars 5 forks source link

Deepspeed Mistral, Memory profiling support and empirical quality checks #168

Closed Anindyadeep closed 2 months ago

Anindyadeep commented 2 months ago

This PR introduces all the changes by PR https://github.com/premAI-io/benchmarks/pull/167 and integrates those in DeepSpeed. DeepSpeed README now has quality checks table for both Llama 2 and Mistral