adam_beta1: 0.9 | |
adam_beta2: 0.999 | |
adam_epsilon: 1.0e-08 | |
adam_weight_decay: 0.01 | |
allow_tf32: false | |
center_crop: false | |
checkpointing_steps: 250 | |
checkpoints_total_limit: null | |
class_data_dir: ./real_reg/samples_penguin | |
class_prompt: penguin | |
concepts_list: null | |
dataloader_num_workers: 2 | |
enable_xformers_memory_efficient_attention: true | |
freeze_model: crossattn_kv | |
gradient_accumulation_steps: 1 | |
gradient_checkpointing: false | |
hflip: true | |
hub_model_id: custom-diffusion-library/penguin | |
hub_token: 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 | |
initializer_token: ktn+pll+ucd | |
instance_data_dir: /grogu/user/nkumari/data_custom_diffusion/penguin/ | |
instance_prompt: photo of a <new1> penguin | |
learning_rate: 1.0e-05 | |
local_rank: 0 | |
logging_dir: logs | |
lr_scheduler: constant | |
lr_warmup_steps: 0 | |
max_grad_norm: 1.0 | |
max_train_steps: 250 | |
mixed_precision: null | |
modifier_token: <new1> | |
num_class_images: 200 | |
num_train_epochs: 1 | |
num_validation_images: 4 | |
output_dir: /grogu/user/nkumari/data_custom_diffusion/diffuser_models/penguin_xformer | |
pretrained_model_name_or_path: CompVis/stable-diffusion-v1-4 | |
prior_generation_precision: null | |
prior_loss_weight: 1.0 | |
push_to_hub: true | |
real_prior: true | |
report_to: tensorboard | |
resolution: 512 | |
resume_from_checkpoint: null | |
revision: null | |
sample_batch_size: 4 | |
scale_lr: true | |
seed: 42 | |
tokenizer_name: null | |
train_batch_size: 2 | |
use_8bit_adam: false | |
validation_epochs: 50 | |
validation_prompt: <new1> penguin sitting on a chair | |
with_prior_preservation: true | |