Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add metadata caching for DreamBooth dataset #1206

Merged
merged 3 commits into from
Mar 26, 2024
Merged

Add metadata caching for DreamBooth dataset #1206

merged 3 commits into from
Mar 26, 2024

Conversation

kohya-ss
Copy link
Owner

This PR is a further development of #1178.

  • Unify cache_metadata and use_cached_meta to cache_info
  • Remove cache_dataset_meta.py because the metadata is cached at the first run.
  • Move cache_info option to DreamBoothSubset because it is not available for FinetuingDataset.

KohakuBlueleaf and others added 3 commits March 24, 2024 15:40
…aiting time for initilization (#1178)

* support meta cached dataset

* add cache meta scripts

* random ip_noise_gamma strength

* random noise_offset strength

* use correct settings for parser

* cache path/caption/size only

* revert mess up commit

* revert mess up commit

* Update requirements.txt

* Add arguments for meta cache.

* remove pickle implementation

* Return sizes when enable cache

---------

Co-authored-by: Kohya S <52813779+kohya-ss@users.noreply.github.com>
@kohya-ss kohya-ss merged commit 78e0a76 into dev Mar 26, 2024
2 checks passed
@kohya-ss kohya-ss deleted the dataset-cache branch March 26, 2024 10:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants