kohya-ss / kohya-ss/sd-scripts

Load latent file without original image

Open
#180 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
7.2k
Forks
1.2k
Avg merge
11m
Merged PRs (30d)
2

Description

Hi,

I'm using a cloud environment to do training.
To minimize the upload file size, I only upload the latent files and the meta_lat.json to the environment, but the current dataset handler does not allow this.

I make the following change to train_util.py:47, and training works fine.
`IMAGE_EXTENSIONS = [".png", ".jpg", ".jpeg", ".webp", ".bmp", ".npz"]`

From my understanding, this change should not break anything.
Please consider this use case in your next update, thx.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by inspecting train_util.py around line 47, where IMAGE_EXTENSIONS is defined, and review how the dataset handler recognizes latent files. Confirm that a dataset containing only latent files and meta_lat.json can be used for training without affecting image-based datasets; no test file is mentioned in the issue.

Written by the indexing model from the issue text.

Assessment

Tech stack
python
Domain
machine-learning
Issue type
Feature
Difficulty
2/5
Estimated time
1-3 hours
Activity status
Stale
Clarity
Clearly specified
Newbie friendliness
52/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.