poc: Rewrite fine_tune.py as train_native.py, tested with sdxl_train.py only

This commit is contained in:
Darren Laurie
2025-02-24 00:05:48 +08:00
parent 894037f2c6
commit 0de1e00f0d
3 changed files with 1981 additions and 874 deletions

View File

@@ -15,6 +15,8 @@ import logging
logger = logging.getLogger(__name__)
ARB_MIN_STEPS = 32
UNET_NUM_BLOCKS_FOR_BLOCK_LR = 23
VAE_SCALE_FACTOR = 0.13025
MODEL_VERSION_SDXL_BASE_V1_0 = "sdxl_base_v1-0"

File diff suppressed because it is too large Load Diff

1758
train_native.py Normal file

File diff suppressed because it is too large Load Diff