PSLD / diffusion-posterior-sampling /configs /imagenet_model_config.yaml
LituRout's picture
add dps config
5fd3a18
# Defaults for image training.
image_size: 256
num_channels: 256
num_res_blocks: 2
channel_mult: ""
learn_sigma: True
class_cond: False
use_checkpoint: False
attention_resolutions: "32,16,8"
num_heads: 4
num_head_channels: 64
num_heads_upsample: -1
use_scale_shift_norm: True
dropout: 0.0
resblock_updown: True
use_fp16: False
use_new_attention_order: False
model_path: models/imagenet256.pt