Pruna AI Files

The juiciest bits 🧃

October was all about Hacktoberfest, and wow, our community really showed up 🥺
From shiny new features to better docs and stronger evals, Pruna got a whole lot more powerful (and prettier) this month.
Let’s dive into what everyone’s been cooking 🔮

---

@R055A leveled up our experimentation game ✨

• feat: tinywikitext dataset by @R055A in https://github.com/PrunaAI/pruna/pull/362
• feat: multi-scale SSIM by @R055A in https://github.com/PrunaAI/pruna/pull/361

---

@kirdmiv brought the vibes to our evaluation module 🎨

• feat: aesthetic laion metric by @kirdmiv in https://github.com/PrunaAI/pruna/pull/424

---

Type checking, but make it sleek 🪶:

• feat: migrate from mypy to ty by @Supakornn in https://github.com/PrunaAI/pruna/pull/360
• style: add jaxtyping annotations and contributor documentation by @vaishnaviparabkar90 @begumcig in https://github.com/PrunaAI/pruna/pull/423

---

@DevManpreet5 was both adding features and fixing our bugs 🧰 :

• feat: add progress bars to EvaluationAgent by @DevManpreet5 in https://github.com/PrunaAI/pruna/pull/348
• docs: fix incorrect clone link in README by @DevManpreet5 in https://github.com/PrunaAI/pruna/pull/347

---

TinyIMDB dataset plus new logging system with adjustable levels by @pranayyb 🎬 :

• feat: Add TinyIMDB dataset for lightweight experiments by @pranayyb in https://github.com/PrunaAI/pruna/pull/374
• feat: add set_logging_level functionality by @pranayyb in https://github.com/PrunaAI/pruna/pull/398

---

@Almonok made sure our repo stays squeaky clean 🛡️:

• feat: add pre-commit trufflehog by @Almonok in https://github.com/PrunaAI/pruna/pull/383

---

DINO has joined our evaluation zoo! Thanks to @Prashankavi @begumcig 🦖:

• feat: dino score by @Prashankavi @begumcig in https://github.com/PrunaAI/pruna/pull/354

---

We have more tutorials, making getting started with Pruna even easier, thanks to @ParagEkbote 🇪🇿:

• Add an End-To-End Tutorial for Efficient-Large-Model/Sana_600M_512px_diffusers by @ParagEkbote in https://github.com/PrunaAI/pruna/pull/322
• Create Compatibility Matrix for Algorithm in Docs by @ParagEkbote in https://github.com/PrunaAI/pruna/pull/403

---

Our very own Pruners were also busy making amazing contributions this month!: 🟣

---

Target modules are extending!:

• feat: add target modules to bnb quantizers by @gsprochette in https://github.com/PrunaAI/pruna/pull/333

---

We have more datasets for experimentation and benchmarking:

• feat: add prompt-only image generation datasets by @nifleisch in https://github.com/PrunaAI/pruna/pull/310
• feat: vbench datamodule by @begumcig in https://github.com/PrunaAI/pruna/pull/397

---

EvaluationAgent now supports multi-GPU inference and multi-GPU latency metrics.

• feat: accelerate inference by @begumcig in https://github.com/PrunaAI/pruna/pull/405

---

More evaluation metric features:

• feat: expose metric information in MetricResult by @sdiazlor in https://github.com/PrunaAI/pruna/pull/326

---

PrunaModel and SmashConfig improvements:

• feat: allow loading models without a smash config by @sharpenb in https://github.com/PrunaAI/pruna/pull/340
• feat: add a get device type function to utils by @simlang in https://github.com/PrunaAI/pruna/pull/416

---

Pruning some bugs 🐞 and maintanence 👩‍🌾:

• docs: removing pruna_pro mentions by @Mel-Alm in https://github.com/PrunaAI/pruna/pull/387
• telemetry: enable metrics and update OTLP endpoint to staging environment by @gtregoat in https://github.com/PrunaAI/pruna/pull/394
• docs: update contributor setup instructions by @johannaSommer in https://github.com/PrunaAI/pruna/pull/355
• docs: add target modules documentation by @gsprochette in https://github.com/PrunaAI/pruna/pull/331
• ci: migrate test models to pruna-test by @gsprochette in https://github.com/PrunaAI/pruna/pull/385
• fix: change vbench dependency to exclude macOS by @begumcig in https://github.com/PrunaAI/pruna/pull/426
• fix: add dataset and inferencer for janus tests by @gsprochette in https://github.com/PrunaAI/pruna/pull/327
• Reduce Flakiness in CI by configuring HF Token and add caching for HF… by @davidberenstein1957 in https://github.com/PrunaAI/pruna/pull/410
• fix: remove outdated prunamodel interface from deploying Sana notebook by @begumcig in https://github.com/PrunaAI/pruna/pull/420

🌱 New faces in the garden

• @DevManpreet5 made their first contribution in https://github.com/PrunaAI/pruna/pull/347
• @R055A made their first contribution in https://github.com/PrunaAI/pruna/pull/362
• @pranayyb made their first contribution in https://github.com/PrunaAI/pruna/pull/374
• @Almonok made their first contribution in https://github.com/PrunaAI/pruna/pull/383
• @Supakornn made their first contribution in https://github.com/PrunaAI/pruna/pull/360
• Our @Mel-Alm made her first contribution 💪 in https://github.com/PrunaAI/pruna/pull/387
• @sharpenb is back with their first contribution since open-sourcing 🚀 in https://github.com/PrunaAI/pruna/pull/340
• @gtregoat made his first contribution too, usually holding it down on the backend, but nice to see him in the main repo! ⚙️ in https://github.com/PrunaAI/pruna/pull/394

Full Changelog: https://github.com/PrunaAI/pruna/compare/v0.2.10...v0.2.11