Commit graph

10 commits

Author SHA1 Message Date
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 b20474f8f5 add various torch.dtype test fixtures 2024-10-03 11:05:09 +02:00
Pierre Chapuis 09af570b23 add DINOv2-FD metric 2024-04-03 16:45:00 +02:00
Laurent 1a8ea9180f refactor dinov2 tests, check against official implementation 2024-04-02 10:02:43 +02:00
Cédric Deltheil 792a0fc3d9 run lint rules using latest isort settings 2023-12-11 11:58:43 +01:00
Benjamin Trom 282578ddc0 add Segment Anything (SAM) to foundational models
Note: dense prompts (i.e. masks) support is still partial (see MaskEncoder)

Co-authored-by: Cédric Deltheil <cedric@deltheil.me>
2023-09-21 11:44:30 +02:00
Pierre Chapuis 32c1cfdbb1 add black to CI 2023-08-30 14:50:03 +02:00
Doryan Kaced 3680f9d196 Add support for learned concepts e.g. via textual inversion 2023-08-28 10:37:39 +02:00
Cédric Deltheil 48f674c433 initial commit 2023-08-04 15:28:41 +02:00