ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-03-13 14:39:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adapter work. Bug fixes. Auto adjust LR when resuming optimizer.

Browse Source
This commit is contained in:
Jaret Burkett
2024-03-17 10:21:47 -06:00
parent 72de68d8aa
commit 016687bda1
8 changed files with 84 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
testing/test_bucket_dataloader.py
View File

@@ -42,6 +42,7 @@ dataset_config = DatasetConfig(
resolution=resolution,
caption_ext='json',
default_caption='default',
clip_image_path='/mnt/Datasets/face_pairs2/control_clean',
buckets=True,
bucket_tolerance=bucket_tolerance,
poi='person',