VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

add fine tuning FLUX.1 (WIP)

Browse Source
This commit is contained in:
Kohya S
2024-08-17 15:36:18 +09:00
parent 7367584e67
commit 400955d3ea
4 changed files with 1007 additions and 162 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
library/train_util.py
View File

@@ -2629,7 +2629,7 @@ class MinimalDataset(BaseDataset):
raise NotImplementedError
def load_arbitrary_dataset(args, tokenizer) -> MinimalDataset:
def load_arbitrary_dataset(args, tokenizer=None) -> MinimalDataset:
module = ".".join(args.dataset_class.split(".")[:-1])
dataset_class = args.dataset_class.split(".")[-1]
module = importlib.import_module(module)