UDiffText / configs /demo.yaml
ZYMPKU's picture
gpu
b9dc102
raw
history blame
510 Bytes
type: "demo"
# path
load_ckpt_path: "./checkpoints/st-step=100000+la-step=100000-simp.ckpt"
model_cfg_path: "./configs/test/textdesign_sd_2.yaml"
# param
H: 512
W: 512
seq_len: 12
batch_size: 1
channel: 4 # AE latent channel
factor: 8 # AE downsample factor
scale: [4.0, 0.0] # content scale, style scale
noise_iters: 10
force_uc_zero_embeddings: ["ref", "label"]
aae_enabled: False
detailed: True
dual_conditioner: False
# runtime
steps: 50
init_step: 0
num_workers: 0
use_gpu: True
gpu: 0
max_iter: 100