Skip to content

[SageMaker] New Flux.2 image generation example#2649

Merged
ehcalabres merged 12 commits into
huggingface:mainfrom
ehcalabres:sagemaker-image-generation
Jul 17, 2026
Merged

[SageMaker] New Flux.2 image generation example#2649
ehcalabres merged 12 commits into
huggingface:mainfrom
ehcalabres:sagemaker-image-generation

Conversation

@ehcalabres

@ehcalabres ehcalabres commented Jul 17, 2026

Copy link
Copy Markdown
Contributor

Description

This PR adds a new SageMaker example about image generation with the Hugging Face vLLM Omni DLC. The topic of the example is: Decorate your house with Flux.2 and Amazon SageMaker AI

It also adds a couple of improvements to the SageMaker docs:

  • Adds the option to include the author and the last modification date for tutorials, as in our other cloud providers docs (i.e. GCP or Microsoft). It also allows to specify the author's Hub handle, and it will add a link to the Hub profile.
  • Notebook assets are replaced by an URL to the respective asset in the huggingface/documentation-images dataset.

Note

Low Risk
Documentation and example-generation scripts only; no production application or auth logic changes.

Overview
Adds a new SageMaker SDK tutorial notebook that deploys FLUX.2-klein-4B on the vLLM-Omni DLC for room redesign (image editing and optional text-to-image), including endpoint setup, chat-completions routing via CustomAttributes, a Gradio interior studio, optional MCP, and cleanup.

Updates the SageMaker docs build so generated example pages show author (with HF profile links from notebook metadata) and last updated (from git history on the source notebook), and rewrites local ./assets / ./imgs image links to the huggingface/documentation-images dataset instead of raw GitHub paths in hub-docs.

Reviewed by Cursor Bugbot for commit 8a665a6. Bugbot is set up for automated code reviews on this repo. Configure here.

@ehcalabres
ehcalabres requested review from alvarobartt and dwarez July 17, 2026 06:37
@HuggingFaceDocBuilderDev

Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Comment thread docs/sagemaker/scripts/auto-update-toctree.py
Comment thread docs/sagemaker/scripts/auto-generate-examples.py
@dwarez

dwarez commented Jul 17, 2026

Copy link
Copy Markdown
Contributor

hey @ehcalabres, this looks great 😄 I ran it end to end and confirm it's working correctly.

I just noticed that image_to_data_uri and resize_to_multiple_of_16 are defined/redefined in the Gradio cell,

@cursor cursor Bot left a comment

Copy link
Copy Markdown

Choose a reason for hiding this comment

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

Cursor Bugbot has reviewed your changes using default effort and found 1 potential issue.

Fix All in Cursor

❌ Bugbot Autofix is OFF. To automatically fix reported issues with cloud agents, enable autofix in the Cursor dashboard.

Reviewed by Cursor Bugbot for commit 9122c0f. Configure here.

@ehcalabres
ehcalabres merged commit 6cfbe03 into huggingface:main Jul 17, 2026
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants