Skip to content

Add an End-To-End Tutorial for Efficient-Large-Model/Sana_600M_512px_diffusers#322

Merged
davidberenstein1957 merged 88 commits intoPrunaAI:mainfrom
ParagEkbote:add-hidream-example
Oct 28, 2025
Merged

Add an End-To-End Tutorial for Efficient-Large-Model/Sana_600M_512px_diffusers#322
davidberenstein1957 merged 88 commits intoPrunaAI:mainfrom
ParagEkbote:add-hidream-example

Conversation

@ParagEkbote
Copy link
Copy Markdown
Contributor

@ParagEkbote ParagEkbote commented Aug 28, 2025

Description

I've reopened the PR which got closed due to it being stale by the stale bot.

Related Issue

Fixes #93

Type of Change

  • Bug fix (non-breaking change which fixes an issue)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • This change requires a documentation update

How Has This Been Tested?

Checklist

  • My code follows the style guidelines of this project
  • I have performed a self-review of my own code
  • I have commented my code, particularly in hard-to-understand areas
  • I have made corresponding changes to the documentation
  • I have added tests that prove my fix is effective or that my feature works
  • New and existing unit tests pass locally with my changes

Additional Notes


Note

Adds a new Jupyter tutorial to optimize Sana diffusers with Pruna and deploy via Hugging Face, and links it from the tutorials index.

  • Docs/Tutorials:
    • New notebook: docs/tutorials/deploying_sana_tutorial.ipynb
      • Optimizes Efficient-Large-Model/Sana_600M_512px_diffusers using Pruna SmashConfig with torchao.
      • Saves smashed model locally and shows optional push to Hugging Face Hub.
      • Loads data-is-better-together/open-image-preferences-v1-binarized with PrunaDataModule for evaluation.
      • Evaluates with EvaluationAgent using TotalTimeMetric and LatencyMetric.
      • Provides a gradio demo app to generate images from prompts.
    • Index update: docs/tutorials/index.rst
      • Adds a grid card linking to deploying_sana_tutorial.ipynb with brief description.

Written by Cursor Bugbot for commit 79f6bab. This will update automatically on new commits. Configure here.

cursor[bot]

This comment was marked as outdated.

cursor[bot]

This comment was marked as outdated.

@ParagEkbote
Copy link
Copy Markdown
Contributor Author

@davidberenstein1957

I have rebased on main and fix any merge conflicts as well. You might need to squash and merge this branch :)

Could you please review?

@davidberenstein1957 davidberenstein1957 changed the base branch from main to fix/flakiness-huggingface October 21, 2025 10:50
@davidberenstein1957 davidberenstein1957 changed the base branch from fix/flakiness-huggingface to main October 21, 2025 10:50
@davidberenstein1957 davidberenstein1957 changed the base branch from main to fix/flakiness-huggingface October 21, 2025 10:52
@davidberenstein1957 davidberenstein1957 changed the base branch from fix/flakiness-huggingface to main October 21, 2025 10:54
Copy link
Copy Markdown
Member

@davidberenstein1957 davidberenstein1957 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good. I think we need a final rebase to rsolve the failing tests :)

Copy link
Copy Markdown
Contributor

@minettekaum minettekaum left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The link to google colab is broken and there was a typo in import torch. Otherwise it looks good!

Copy link
Copy Markdown
Contributor

@minettekaum minettekaum left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm :)

@ParagEkbote
Copy link
Copy Markdown
Contributor Author

Is this PR ready to be merged? Could you please let me know?

cc: @davidberenstein1957

@davidberenstein1957 davidberenstein1957 merged commit 95b2fc4 into PrunaAI:main Oct 28, 2025
4 checks passed
@ParagEkbote ParagEkbote deleted the add-hidream-example branch October 30, 2025 07:22
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[FEATURE] write tutorial on optimizing HiDream-ai/HiDream-I1-Dev

4 participants