Upload json files

model/args.json

@@ -0,0 +1,66 @@
+{
+  "pretrained_model_name_or_path": "stabilityai/stable-diffusion-2",
+  "pretrained_vae_name_or_path": "stabilityai/sd-vae-ft-mse",
+  "revision": "fp16",
+  "tokenizer_name": null,
+  "instance_data_dir": null,
+  "class_data_dir": null,
+  "instance_prompt": null,
+  "class_prompt": null,
+  "save_sample_prompt": "train with fighter jet in the sky",
+  "save_sample_negative_prompt": null,
+  "n_save_sample": 4,
+  "save_guidance_scale": 7.5,
+  "save_infer_steps": 20,
+  "pad_tokens": false,
+  "with_prior_preservation": true,
+  "prior_loss_weight": 1.0,
+  "num_class_images": 150,
+  "output_dir": "/content/stable_diffusion_weights/ldg_scn",
+  "seed": 777,
+  "resolution": 512,
+  "center_crop": false,
+  "train_text_encoder": true,
+  "train_batch_size": 1,
+  "sample_batch_size": 4,
+  "num_train_epochs": 3,
+  "max_train_steps": 800,
+  "gradient_accumulation_steps": 1,
+  "gradient_checkpointing": false,
+  "learning_rate": 1e-06,
+  "scale_lr": false,
+  "lr_scheduler": "constant",
+  "lr_warmup_steps": 80,
+  "use_8bit_adam": true,
+  "adam_beta1": 0.9,
+  "adam_beta2": 0.999,
+  "adam_weight_decay": 0.01,
+  "adam_epsilon": 1e-08,
+  "max_grad_norm": 1.0,
+  "push_to_hub": false,
+  "hub_token": null,
+  "hub_model_id": null,
+  "logging_dir": "logs",
+  "log_interval": 10,
+  "save_interval": 800,
+  "save_min_steps": 0,
+  "mixed_precision": "fp16",
+  "not_cache_latents": false,
+  "hflip": false,
+  "local_rank": -1,
+  "concepts_list": [
+    {
+      "instance_prompt": "train",
+      "class_prompt": "photo of a train",
+      "instance_data_dir": "/content/data/trains",
+      "class_data_dir": "/content/data/class_trains"
+    },
+    {
+      "instance_prompt": "jets",
+      "class_prompt": "fighter jets in the sky",
+      "instance_data_dir": "/content/data/fighter_jets",
+      "class_data_dir": "/content/data/jets"
+    }
+  ],
+  "read_prompts_from_txts": false
+}

model/model_index.json

@@ -0,0 +1,34 @@
+{
+  "_class_name": "StableDiffusionPipeline",
+  "_diffusers_version": "0.22.0.dev0",
+  "_name_or_path": "stabilityai/stable-diffusion-2",
+  "feature_extractor": [
+    null,
+    null
+  ],
+  "requires_safety_checker": false,
+  "safety_checker": [
+    null,
+    null
+  ],
+  "scheduler": [
+    "diffusers",
+    "DDIMScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "CLIPTextModel"
+  ],
+  "tokenizer": [
+    "transformers",
+    "CLIPTokenizer"
+  ],
+  "unet": [
+    "diffusers",
+    "UNet2DConditionModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKL"
+  ]
+}