Laurent
316fe6e4f0
update tests to use new fixtures
2024-10-14 15:12:59 +02:00
Laurent
94eeb1afc3
add new weight related test fixtures
2024-10-14 15:12:59 +02:00
Pierre Chapuis
2796117d2d
properly check for bfloat16
...
- we check only the test device, not the machine in general
- we don't want emulated bfloat16 (e.g. CPU)
2024-10-09 11:20:27 +02:00
Laurent
f3d2b6c325
modify some foundational tests to also test in float16 and bfloat16
2024-10-03 11:05:09 +02:00
Laurent
b20474f8f5
add various torch.dtype test fixtures
2024-10-03 11:05:09 +02:00
Laurent
4360aa046f
add test_diffusion_std_random_init_bfloat16 e2e test
2024-10-03 11:05:09 +02:00
Pierre Chapuis
ae5ef4f14c
make sure adapters can access the context of their parent chain
...
https://github.com/finegrain-ai/refiners/issues/456
2024-10-03 10:51:11 +02:00
Laurent
f89c4f720d
add test_multi_upscaler_small to check that MultiUpscaler works with small (16x16px) images
2024-10-01 11:14:52 +02:00
Laurent
7b6f97347e
update some reference test images
2024-10-01 11:09:57 +02:00
Laurent
5a92285104
add torch.Generator to MultiUpscaler.upscale + make MultiUpscaler.diffuse_targets "stateless"
CI / lint_and_typecheck (push) Has been cancelled
Deploy docs to GitHub Pages / Deploy docs (push) Has been cancelled
Spell checker / Spell check (push) Has been cancelled
2024-09-26 11:31:00 +02:00
Pierre Chapuis
ed7e2e5964
check timesteps stay the same in bfloat16
2024-09-26 08:55:07 +02:00
Pierre Chapuis
f1c8503e68
use default (1e-8) atol for basic DPM solver (used in SD1.5)
2024-09-26 08:55:07 +02:00
Pierre Chapuis
83b931296f
fix DPM-Solver with bfloat16
CI / lint_and_typecheck (push) Waiting to run
Deploy docs to GitHub Pages / Deploy docs (push) Waiting to run
Spell checker / Spell check (push) Waiting to run
2024-09-25 14:41:19 +02:00
Pierre Chapuis
73089a4e2d
fix typing issue in dinov2 test
CI / lint_and_typecheck (push) Has been cancelled
Deploy docs to GitHub Pages / Deploy docs (push) Has been cancelled
Spell checker / Spell check (push) Has been cancelled
2024-09-18 10:44:24 +02:00
Laurent
2c0174f50e
rename test_unet.py to test_sd15_unet.py + use test_device fixture
CI / lint_and_typecheck (push) Has been cancelled
Deploy docs to GitHub Pages / Deploy docs (push) Has been cancelled
Spell checker / Spell check (push) Has been cancelled
2024-09-10 09:16:22 +02:00
Laurent
444882a734
move some tests into the adapters test folder
CI / lint_and_typecheck (push) Waiting to run
Deploy docs to GitHub Pages / Deploy docs (push) Waiting to run
Spell checker / Spell check (push) Waiting to run
2024-09-09 17:44:26 +02:00
Laurent
6d58492097
loosen the tolerances for some sam tests (because of a recent pytorch upgrade)
2024-09-09 17:17:12 +02:00
limiteinductive
af6c5aecbe
add karras sigmas to dpm solver
2024-09-06 15:30:02 +02:00
ily-R
277b0fd837
ella adapter implementation. tested with sd1.5 model
2024-09-04 11:38:22 +02:00
Pierre Chapuis
7ca1774b5f
add box segmenter solution
CI / lint_and_typecheck (push) Has been cancelled
Deploy docs to GitHub Pages / Deploy docs (push) Has been cancelled
Spell checker / Spell check (push) Has been cancelled
2024-08-30 09:26:53 +02:00
Pierre Chapuis
e643006887
fix .to for MVANet
CI / lint_and_typecheck (push) Waiting to run
Deploy docs to GitHub Pages / Deploy docs (push) Waiting to run
Spell checker / Spell check (push) Waiting to run
2024-08-27 18:41:29 +02:00
Pierre Chapuis
10dfa73a09
Add Multi-View Aggregation Network (MVANet)
...
Co-authored-by: Pierre Colle <piercus@gmail.com>
2024-08-26 13:59:02 +02:00
limiteinductive
51dcd7772c
implement foreground conditioned ic light
2024-08-12 12:09:23 +02:00
limiteinductive
1de567590b
fix typing issues coming from torch 2.4 version ; typing is not guaranteed for torch < 2.4
2024-08-02 12:02:00 +02:00
limiteinductive
09a9dfd494
Add stochastic sampling to DPM solver (SDE)
CI / lint_and_typecheck (push) Waiting to run
Deploy docs to GitHub Pages / Deploy docs (push) Waiting to run
Spell checker / Spell check (push) Waiting to run
2024-07-23 11:13:12 +02:00
Pierre Chapuis
daee77298d
improve FrankenSolver
...
It now takes a Scheduler factory instead of a Scheduler.
This lets the user potentially recreate the Scheduler on `rebuild`.
It also properly sets the device and dtype on rebuild,
and it has better typing.
2024-07-19 16:46:52 +02:00
Laurent
f44ae150a7
improve/add MultiDiffusion and MultiUpscaler e2e tests
...
Co-authored-by: limiteinductive <benjamin@lagon.tech>
Co-authored-by: Cédric Deltheil <355031+deltheil@users.noreply.github.com>
2024-07-11 15:23:02 +02:00
Pierre Chapuis
b4db08de24
use "solver" (not scheduler) wording in tests
2024-07-10 19:31:34 +02:00
Pierre Chapuis
9e8c2a3753
add FrankenSolver
...
This solver is designed to use Diffusers Schedulers as Refiners Solvers.
2024-07-10 19:31:34 +02:00
limiteinductive
b42881e54e
Implement Tiled Autoencoder inference to save VRAM
2024-06-26 11:59:18 +02:00
limiteinductive
b16cb7fbb6
Add ControlNet Tile e2e test
2024-06-25 09:27:08 +02:00
Pierre Chapuis
81f77e7fb0
update doc examples test references
2024-06-24 17:19:05 +02:00
Pierre Chapuis
9be3580ab3
avoid OOMs in SAM tests
2024-06-24 17:19:05 +02:00
Pierre Chapuis
91c0fd24d8
ruff: reorder test imports
2024-06-24 17:19:05 +02:00
Pierre Chapuis
5131212bbb
update lightning test references
2024-06-24 17:19:05 +02:00
Pierre Chapuis
8366861aed
update IP adapter test references
2024-06-24 17:19:05 +02:00
Pierre Chapuis
2ba83f575e
switch pytest import mode to importlib
...
see:
https://docs.pytest.org/en/7.1.x/explanation/goodpractices.html#choosing-an-import-mode
https://docs.pytest.org/en/7.1.x/explanation/pythonpath.html#import-modes
This change fixes the SAM tests.
2024-06-24 17:19:05 +02:00
limiteinductive
15ccdb38f3
Add scale_decay parameter for SD1 ControlNet
2024-06-24 13:21:27 +02:00
limiteinductive
3a7f14e4dc
Fix clock log order ; fix that the first iteration was skipped
2024-05-21 17:57:14 +02:00
limiteinductive
0bec9a855d
annotated validators for TimeValue
2024-05-09 10:53:58 +02:00
limiteinductive
22f4f4faf1
DataLoader validation
2024-05-09 10:53:58 +02:00
limiteinductive
38bddc49bd
implement data_iterable
2024-05-09 10:53:58 +02:00
limiteinductive
d6c225a112
implement data_iterable (bis)
2024-05-09 10:53:58 +02:00
limiteinductive
de8334b6fc
remove dataset length
2024-05-09 10:53:58 +02:00
limiteinductive
b497b27cd3
remove dataset length (bis)
2024-05-09 10:53:58 +02:00
Benjamin Trom
1db0845db2
update test_trainer.py
2024-05-09 10:53:58 +02:00
limiteinductive
603c8abb1e
fix clock
2024-05-09 10:53:58 +02:00
limiteinductive
44760ac19f
deprecate evaluation
2024-05-09 10:53:58 +02:00
limiteinductive
061d44888f
batch to step
2024-05-09 10:53:58 +02:00
limiteinductive
b7bb8bba80
remove EventConfig
...
This is a partial rollback of commit 5dde281
2024-05-09 10:53:58 +02:00