CoolFace
Modelpublic

atsyplikhin/rita_sd_model

sourceHugging Faceupdated 3y agoView on Hugging Face
0likes8downloads
args.json58 linesDownload Raw Back to root
1{2  "pretrained_model_name_or_path": "CompVis/stable-diffusion-v1-4",3  "pretrained_vae_name_or_path": null,4  "revision": "fp16",5  "tokenizer_name": null,6  "instance_data_dir": "Rita-style",7  "class_data_dir": null,8  "instance_prompt": "RitaT style collage",9  "class_prompt": null,10  "save_sample_prompt": null,11  "save_sample_negative_prompt": null,12  "n_save_sample": 4,13  "save_guidance_scale": 7.5,14  "save_infer_steps": 50,15  "pad_tokens": false,16  "with_prior_preservation": false,17  "prior_loss_weight": 1.0,18  "num_class_images": 100,19  "output_dir": "model_rita1",20  "seed": null,21  "resolution": 512,22  "center_crop": false,23  "train_text_encoder": false,24  "train_batch_size": 1,25  "sample_batch_size": 4,26  "num_train_epochs": 32,27  "max_train_steps": 1600,28  "gradient_accumulation_steps": 1,29  "gradient_checkpointing": true,30  "learning_rate": 5e-06,31  "scale_lr": false,32  "lr_scheduler": "constant",33  "lr_warmup_steps": 0,34  "use_8bit_adam": true,35  "adam_beta1": 0.9,36  "adam_beta2": 0.999,37  "adam_weight_decay": 0.01,38  "adam_epsilon": 1e-08,39  "max_grad_norm": 1.0,40  "push_to_hub": false,41  "hub_token": null,42  "hub_model_id": null,43  "logging_dir": "logs",44  "log_interval": 10,45  "save_interval": 10000,46  "save_min_steps": 0,47  "mixed_precision": "no",48  "not_cache_latents": false,49  "local_rank": -1,50  "concepts_list": [51    {52      "instance_prompt": "RitaT style collage",53      "class_prompt": null,54      "instance_data_dir": "Rita-style",55      "class_data_dir": null56    }57  ]58}