Skip to content

Add images api#3

Open
jglee-sqbits wants to merge 21 commits into
add-flux1-pipelinefrom
add-images-api
Open

Add images api#3
jglee-sqbits wants to merge 21 commits into
add-flux1-pipelinefrom
add-images-api

Conversation

@jglee-sqbits
Copy link
Copy Markdown
Collaborator

@jglee-sqbits jglee-sqbits commented Jan 19, 2026

Description

This PR introduces the Diffusion pipeline along with the serving infrastructure. It covers the entrypoint setup and image request handling.

Testing:

  • Run the command below to verify the changes:
  • Run with pipelines_diffusion for torch dependency(cudnn fallback)
./bazelw run //max/python/max/entrypoints:pipelines_diffusion -- \
  images generate \
  --output /absolute/output/image/path \
  --prompt "a cat in the jungle" \
  --model black-forest-labs/FLUX.1-dev \
  --seed 4242
./bazelw run //max/python/max/entrypoints:pipelines_diffusion -- serve --model black-forest-labs/FLUX.1-dev --task image_generation

curl test

python max/examples/diffusion/client_example.py 

openai api test

python max/examples/diffusion/openai_api_example.py 

Key Changes:

  • Diffusion Pipeline: Added the diffusion pipeline with queue integration.

  • Serve: Implemented the serve logic.

  • Entrypoint: Added the application entrypoint.

  • Image Processing:

    • Added Image Request handling.

    • Added Image Pipeline.

⚠️ Important Notes:

  • WIP (Work In Progress): This PR will be maintained and updated until the serve implementation is fully complete. It will be merged afterwards.

  • Examples: Please note that the provided examples might not work as expected in this current state.

@kkimmk kkimmk force-pushed the add-flux1-pipeline branch from a149668 to c16451a Compare February 3, 2026 13:46
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.

2 participants