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
Laurent
7aff743019
initialize StableDiffusion_1_Inpainting with a 9 channel SD1Unet if not provided
2024-04-23 16:58:22 +02:00
limiteinductive
5dde281ada
Implement EventConfig
2024-04-22 18:14:33 +02:00
limiteinductive
446796da57
Refactor TimeValue
2024-04-18 20:58:47 +02:00
Laurent
17246708b9
Add sample_noise
staticmethod and modify add_noise
to support batched steps
2024-04-18 12:55:49 +02:00
limiteinductive
7427c171f6
fix training_utils requirements check
2024-04-17 18:10:28 +02:00
Pierre Colle
bf7852b88e
SAM: image_to_scaled_tensor gray images
2024-04-16 18:45:17 +02:00
limiteinductive
f48712ee29
lint
2024-04-16 16:24:33 +02:00
limiteinductive
347fdbc794
add init file for segment_anything tests
2024-04-16 16:24:33 +02:00
limiteinductive
b9b999ccfe
turn scoped_seed into a context manager
2024-04-13 15:03:35 +02:00
Pierre Colle
64692c3b5b
TrainerClock: assert dataset_length >= batch_size
2024-04-12 15:05:52 +02:00
Pierre Colle
0ac290f67d
SAM: expose sizing helpers
2024-04-12 08:56:23 +02:00
Laurent
06ff2f0a5f
add support for dinov2 giant flavors
2024-04-11 14:48:33 +02:00
limiteinductive
f26b6ee00a
add static typing to __call__ method for latent_diffusion models ; fix multi_diffusion bug that wasn't taking guidance_scale into account
2024-04-11 12:13:30 +02:00
Pierre Chapuis
09af570b23
add DINOv2-FD metric
2024-04-03 16:45:00 +02:00
Laurent
2ecf7e4b8c
skip dinov2 float16 test on cpu + test dinov2 when batch_size>1
2024-04-02 18:57:25 +02:00
Pierre Chapuis
fd5a15c7e0
update pyright and fix Pillow 10.3 typing issues
2024-04-02 18:15:52 +02:00
Laurent
1a8ea9180f
refactor dinov2 tests, check against official implementation
2024-04-02 10:02:43 +02:00
Pierre Colle
6c37e3f933
hq-sam : weights/load_weights
2024-03-29 11:25:43 +01:00
Laurent
7e64ba4011
modify ip_adapter's CrossAttentionAdapters injection logic
2024-03-26 11:15:04 +01:00
Pierre Colle
cba83b0558
SAM init with mask_decoder after #325
2024-03-24 20:18:57 +01:00
Pierre Colle
5c937b184a
HQ-SAM logit equal test, following #331
2024-03-23 21:58:32 +01:00
Pierre Colle
2763db960e
SAM e2e test tolerance explained
2024-03-22 21:31:28 +01:00
Pierre Chapuis
364e196874
support no CFG in compute_clip_text_embedding
2024-03-22 17:06:51 +01:00
hugojarkoff
a93ceff752
Add HQ-SAM Adapter
2024-03-21 15:36:55 +01:00
hugojarkoff
c6b5eb24a1
Add logits comparison for base SAM in single mask output prediction mode
2024-03-21 10:48:48 +01:00
limiteinductive
38c86f59f4
Switch gradient clipping to native torch torch.nn.utils.clip_grad_norm_
2024-03-19 22:08:48 +01:00
Pierre Colle
68fe725767
Add multimask_output flag to SAM
2024-03-19 17:40:26 +01:00
limiteinductive
6a72943ff7
change TimeValue to a dataclass
2024-03-19 14:49:24 +01:00
Pierre Chapuis
5d784bedab
add test for "Adapting SDXL" guide
2024-03-08 15:43:57 +01:00
Pierre Chapuis
72fa17df48
fix slider loras test
2024-03-08 15:43:57 +01:00
Pierre Chapuis
8c7fcbc00f
LoRA manager: move exclude / include to add_loras call
...
Always exclude the TimestepEncoder by default.
This is because some keys include both e.g. `resnet` and `time_emb_proj`.
Preprocess blocks that tend to mix up with others in a separate
auto_attach call.
2024-03-08 15:43:57 +01:00
Pierre Chapuis
052a20b897
remove add_multiple_loras
2024-03-08 15:43:57 +01:00
Pierre Chapuis
c383ff6cf4
fix DPO LoRA loading in tests
2024-03-08 15:43:57 +01:00
Pierre Chapuis
1eb71077aa
use same scale setter / getter interface for all controls
2024-03-08 11:29:28 +01:00
Pierre Chapuis
be2368cf20
ruff 3 formatting (Rye 0.28)
2024-03-08 10:42:05 +01:00
Pierre Chapuis
a0be5458b9
snip long prompt in tests
2024-03-05 19:54:44 +01:00