mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-10-12 02:17:16 +00:00
kohya-ssandClaude Fable 5.1
bbf63c6943
Preparation for the transformers 5.x / diffusers 0.40 upgrade. The harness records the outputs of the code paths that depend on those libraries with the currently pinned versions and compares them after the upgrade: - text encoders of every model family (SD1/2, SDXL, SD3, FLUX.1, Lumina, HunyuanImage, Anima) through the trainers' own TokenizeStrategy / TextEncodingStrategy and family loaders, for a fixed prompt set (empty, short, tag list, Japanese + emoji, >77 tokens, >225 tokens), per prompt and batched; token ids / masks must match exactly, float outputs are compared with per-model tolerances and max abs / rel diff and cosine similarity are printed for every array - diffusers AutoencoderKL encode / decode of a fixed image for SD / SDXL - the diffusers noise schedulers used for sample generation (no weights needed): timesteps, alphas_cumprod, sigmas and a few step() calls Model paths live in tests/local/models.toml (git-ignored; template in models.example.toml), references in tests/local/references/ (git-ignored). `pytest` from the repository root does not descend into tests/local (norecursedirs); run `python tests/local/regression_te.py record|compare` or `pytest tests/local` explicitly. Co-Authored-By: Claude Fable 5.1 <[email protected]>
12 lines
263 B
INI
12 lines
263 B
INI
[pytest]
|
|
minversion = 6.0
|
|
testpaths =
|
|
tests
|
|
# tests/local needs local model weights and is run explicitly: pytest tests/local
|
|
norecursedirs = local
|
|
filterwarnings =
|
|
ignore::DeprecationWarning
|
|
ignore::UserWarning
|
|
ignore::FutureWarning
|
|
pythonpath = .
|