Skip to content

[tests] implement base model output caching in model-level tests#14059

Open
sayakpaul wants to merge 4 commits into
mainfrom
cache-base-model-output
Open

[tests] implement base model output caching in model-level tests#14059
sayakpaul wants to merge 4 commits into
mainfrom
cache-base-model-output

Conversation

@sayakpaul

@sayakpaul sayakpaul commented Jun 24, 2026

Copy link
Copy Markdown
Member

Implements fixture-level caching of base model output to avoid recomputation:

  • common.py
  • memory.py

Other mixins don't have the right contract to implement it yet.

@github-actions github-actions Bot added tests size/M PR with diff < 200 LOC labels Jun 24, 2026
@sayakpaul

Copy link
Copy Markdown
Member Author

/diffusers-bot pytest tests/models/transformers/test_models_transformer_flux.py -k "offload"

@github-actions

github-actions Bot commented Jun 24, 2026

Copy link
Copy Markdown
Contributor

pytest tests/models/transformers/test_models_transformer_flux.py -k "offload" passed on GPU — view logs.

@sayakpaul sayakpaul requested review from DN6 and dg845 and removed request for DN6 June 24, 2026 20:41
@sayakpaul sayakpaul marked this pull request as ready for review June 24, 2026 20:41
@sayakpaul sayakpaul requested a review from DN6 June 24, 2026 20:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

size/M PR with diff < 200 LOC tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant