This commit is contained in:
Darren Laurie
2025-04-06 22:29:36 +08:00
parent 03c82ae263
commit 4b21fab176
2 changed files with 2 additions and 1 deletions

View File

@@ -2191,7 +2191,7 @@ class FineTuningDataset(BaseDataset):
validation_seed: int,
validation_split: float,
skip_npz_check: Optional[bool] = SKIP_NPZ_PATH_CHECK,
resize_interpolation: Optional[str],
resize_interpolation: Optional[str] = None,
) -> None:
super().__init__(resolution, network_multiplier, debug_dataset, resize_interpolation)

View File

@@ -15,6 +15,7 @@ tensorboard
safetensors
torchao
pytorch-optimizer
deepspeed==0.15.4
# gradio==3.16.2
altair
easygui