Commit graph

468 commits

Author SHA1 Message Date
Cédric Deltheil 3a9cd95615 docs.yml: revamp workflow
Adapted from the Material for MkDocs template[1]. The former workflow
was using mkdocs-deploy-gh-pages[2] which runs in an Alpine-based Docker
container preventing from installing PyTorch (needed for mkdocstrings).

[1]:
https://squidfunk.github.io/mkdocs-material/publishing-your-site/#with-github-actions
[2]:
https://github.com/mhausenblas/mkdocs-deploy-gh-pages
2024-02-01 15:45:45 +01:00
Cédric Deltheil df45b926aa requirements.docs.txt: add refiners[training]
Since mkdocstrings must access it.
2024-02-01 12:32:05 +01:00
Laurent 9308612f46 add mkdocs-literate-nav in requirements.docs.txt 2024-02-01 11:48:32 +01:00
Laureηt 17fda69e1a fix typo in docs/reference/SUMMARY.md
Co-authored-by: Cédric Deltheil <355031+deltheil@users.noreply.github.com>
2024-02-01 11:48:32 +01:00
Laurent 099510d6c0 configure and initialize mkdocstrings 2024-02-01 11:48:32 +01:00
Cédric Deltheil b94706ff79 ci.yml: specify including paths 2024-02-01 11:20:07 +01:00
Pierre Chapuis ab63a01582 new docs scaffold 2024-01-31 18:23:43 +01:00
limiteinductive abe50076a4 add NoiseSchedule to solvers __init__ + simplify some import pathing
further improve import pathing
2024-01-31 17:03:52 +01:00
limiteinductive 73f6ccfc98 make Scheduler a fl.Module + Change name Scheduler -> Solver 2024-01-31 17:03:52 +01:00
Cédric Deltheil 07cb2ff21c gh-pages: update CUSTOM_DOMAIN env variable
docs.refine.rs -> refine.rs
2024-01-31 16:10:43 +01:00
Laurent e6f5a031d9 add plugins, markdown_extensions, theme.features settings in mkdocs.yml 2024-01-31 15:41:19 +01:00
Pierre Chapuis b2c26ab51c drop scipy dependency (now in bitsandbytes) 2024-01-31 15:03:06 +01:00
Pierre Chapuis 7eb8eb4c68 add support for pytorch 2.2 (2.1 is still supported)
also bump all dev dependencies to their latest version
2024-01-31 15:03:06 +01:00
Cédric Deltheil 45357c5548 mkdocs.yml: add mkdocstrings to plugins 2024-01-31 14:27:47 +01:00
Cédric Deltheil 146a1fae2a add mkdocstrings as dev dependency
Via:

    rye add --dev "mkdocstrings[python]"

Also added to requirements.docs.txt for GH Pages.
2024-01-31 14:27:47 +01:00
Cédric Deltheil 5634e68fde add end-to-end test for multi-ip adapter 2024-01-31 11:03:49 +01:00
Cédric Deltheil ca5c5a7ca5 add helper for multiple image prompts 2024-01-31 11:03:49 +01:00
Cédric Deltheil fd01ba910e fix minor typos in code and docs 2024-01-30 09:52:40 +01:00
Cédric Deltheil feff4c78ae segment-anything: fix class name typo
Note: weights are impacted
2024-01-30 09:52:40 +01:00
Cédric Deltheil 64b52b407f mkdocs.yml: fix typo 2024-01-30 09:52:40 +01:00
Pierre Chapuis 1eb5c779de better coverage config
- exclude e.g. tests
- include files not imported at all
- exclude some common patterns (see https://coverage.readthedocs.io/en/7.4.1/excluding.html#excluding-source-files)
2024-01-29 22:32:16 +01:00
Pierre Chapuis 19bc081658 call GC between each e2e test to avoid OOM 2024-01-29 17:57:40 +01:00
Cédric Deltheil 5a8389ee0d pyproject.toml: formatting 2024-01-29 15:10:06 +01:00
Cédric Deltheil cd64bb29e6 add rye scripts for code coverage 2024-01-29 15:10:06 +01:00
Cédric Deltheil 1efb7a0e10 rye add --dev coverage 2024-01-29 15:10:06 +01:00
Cédric Deltheil c01dad1c77 requirements.lock: update due to rye 0.19.0
i.e. newly added `lock --with-sources` option
2024-01-29 15:10:06 +01:00
Pierre Chapuis deb5e930ae fix exclusions for Downsample and Upsample
(also simplify list comprehension for exclusion list)
2024-01-29 11:11:14 +01:00
Pierre Chapuis 83c95fcf44 fix sorting method for LoRA keys
- support _out_0
- sort _in before _out
- avoid false positives by only considering suffixes
2024-01-29 11:11:14 +01:00
Pierre Chapuis ce22c8f51b fix detection of unet-only LoRAs 2024-01-29 11:11:14 +01:00
Pierre Chapuis ce0339b4cc add a get_path helper to modules 2024-01-26 19:31:13 +01:00
limiteinductive 0ee2d5e075 Fix warmup steps calculation when gradient_accumulation is used 2024-01-25 12:20:36 +01:00
Bryce 12a5439fc4 refactor: rename noise => predicted_noise
and in euler, `alt_noise` can now be simply `noise`
2024-01-24 18:15:10 +01:00
Cédric Deltheil 695c24dd3a image_prompt: remove obsolete comment
Not needed anymore since #168 (CrossAttentionAdapter refactoring)
2024-01-24 09:44:00 +01:00
limiteinductive 3b458f0d8d fix test_names LoraManager test 2024-01-23 14:12:03 +01:00
limiteinductive 421da6a3b6 Load Multiple LoRAs with SDLoraManager 2024-01-23 14:12:03 +01:00
Pierre Chapuis fb2f0e28d4 add rebuild() to Scheduler interface
for use in `set_inference_steps()`
2024-01-23 11:11:50 +01:00
Pierre Chapuis a5c665462a add missing constructor arguments to DDPM scheduler 2024-01-23 11:11:50 +01:00
Cédric Deltheil 40c33b9595 rollback to 50 inference steps in IP-Adapter tests
Follow up of 8a36c8c

This is what is used in the official notebook (ip_adapter_demo.ipynb and
ip_adapter-plus_demo.ipynb)
2024-01-22 09:35:31 +01:00
limiteinductive ed3621362f Add load_tensors utils in fluxion 2024-01-21 12:34:33 +01:00
Pierre Colle 91aea9b7ff fix: summarize_tensor(tensor) when tensor.numel() == 0 2024-01-20 14:32:35 +01:00
Cédric Deltheil 2b4bc77534 prepare_test_weights: update segment-anything hash
Follow up of 00f494e
2024-01-20 14:05:45 +01:00
Pierre Chapuis f6beee8388 tweak docstring for DDPM 2024-01-19 19:01:02 +01:00
Pierre Chapuis 8a36c8c279 make the first diffusion step a first class property of LDM & Schedulers 2024-01-19 18:52:45 +01:00
hugojarkoff 42b7749630 Fix references for e2e tests 2024-01-19 15:00:03 +01:00
Pierre Chapuis de6266010d fix typo (sinuosoidal -> sinusoidal) 2024-01-19 14:37:45 +01:00
Pierre Chapuis d34e36797b fix typo (sinuosidal -> sinusoidal) 2024-01-19 14:27:37 +01:00
Cédric Deltheil fde61757fb summarize_tensor: fix minor warning
Calling `tensor.float()` on a complex tensor raises a warning:

    UserWarning: Casting complex values to real discards the imaginary
    part (Triggered internally at ../aten/src/ATen/native/Copy.cpp:299.)

Follow up of #171
2024-01-19 11:34:47 +01:00
Pierre Chapuis aa9f572611 training dependencies are test dependencies 2024-01-19 10:09:33 +01:00
Pierre Chapuis ce3035923b improve DPM solver test 2024-01-18 19:23:11 +01:00
Pierre Chapuis 999e429697 fix bug in dpm_solver_first_order_update 2024-01-18 19:23:11 +01:00