41 lines
803 B
YAML
41 lines
803 B
YAML
steps: 50000
|
|
batch: 128
|
|
fconst: 30000
|
|
cosine: True
|
|
resume: False
|
|
device: "cuda"
|
|
p_loss_weight: 1.0
|
|
r_loss_weight: 0.0
|
|
b_loss_weight: 0.0
|
|
h_loss_weight: 0.0
|
|
save_image: 1000
|
|
eval_model: 10
|
|
|
|
model:
|
|
src: xl # Stable Diffusion Extra Large
|
|
dst: v1 # Stable Diffusion 1.x
|
|
rev: "v4.0-rc16"
|
|
args:
|
|
scale: 1.0
|
|
ch_in: 4
|
|
ch_out: 4
|
|
ch_mid: 64
|
|
blocks: 12
|
|
|
|
optim:
|
|
lr: 5.0e-4
|
|
beta1: 0.5
|
|
beta2: 0.95
|
|
|
|
dataset:
|
|
src: "./latents/xl_256px_combined.bin"
|
|
dst: "./latents/v1_256px_combined.bin"
|
|
preload: False
|
|
evals:
|
|
main:
|
|
src: "./latents/test_eru/test_xl_768px.npy"
|
|
dst: "./latents/test_eru/test_v1_768px.npy"
|
|
aux:
|
|
src: "./latents/test_bga/test_xl_768px.npy"
|
|
dst: "./latents/test_bga/test_v1_768px.npy"
|