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