Upgrade to SD2

.gitattributes

@@ -32,4 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
-output.png filter=lfs diff=lfs merge=lfs -text
+output.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -12,15 +12,10 @@ inference: true
 
 This the Stable Diffusion model fine-tuned the Low Poly Office concept taught to Stable Diffusion with Dreambooth.
 It can be used by modifying the `instance_prompt`: **a photo of lowpoly_office**
-
 # Run on [Mirage](https://app.mirageml.com)
 Run this model and explore text-to-3D on [Mirage](https://app.mirageml.com)!
-
 Here are is a sample output for this model:
 ![image 0](https://huggingface.co/MirageML/lowpoly-office/resolve/main/output.png)
-
 # Share your Results and Reach us on [Discord](https://discord.gg/9B2Pu2bEvj)!
-
 [![Discord Server](https://discord.com/api/guilds/1022387303022338058/widget.png?style=banner2)](https://discord.gg/9B2Pu2bEvj)
-
 [Image Source](https://www.behance.net/search/images?similarStyleImagesId=723122315)

feature_extractor/preprocessor_config.json

@@ -1,20 +0,0 @@
-{
-  "crop_size": 224,
-  "do_center_crop": true,
-  "do_convert_rgb": true,
-  "do_normalize": true,
-  "do_resize": true,
-  "feature_extractor_type": "CLIPFeatureExtractor",
-  "image_mean": [
-    0.48145466,
-    0.4578275,
-    0.40821073
-  ],
-  "image_std": [
-    0.26862954,
-    0.26130258,
-    0.27577711
-  ],
-  "resample": 3,
-  "size": 224
-}

logs/dreambooth/1669594587.0222032/events.out.tfevents.1669594587.instance-3.1963.1

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:3042d5939982296c9a66f8986270f0e350dc18035ee4f6b93fa530b9fc0992f8
-size 1908

logs/dreambooth/1669594587.0388916/hparams.yml

@@ -1,37 +0,0 @@
-adam_beta1: 0.9
-adam_beta2: 0.999
-adam_epsilon: 1.0e-08
-adam_weight_decay: 0.01
-center_crop: false
-class_data_dir: null
-class_prompt: null
-gradient_accumulation_steps: 1
-gradient_checkpointing: true
-hub_model_id: null
-hub_token: null
-instance_data_dir: /home/diffusers/examples/dreambooth/datasets/3-lowpoly/office/class
-instance_prompt: a photo of lowpoly_office
-learning_rate: 2.0e-06
-local_rank: -1
-logging_dir: logs
-lr_scheduler: constant
-lr_warmup_steps: 0
-max_grad_norm: 1.0
-max_train_steps: 2000
-mixed_precision: null
-num_class_images: 100
-num_train_epochs: 223
-output_dir: /home/diffusers/examples/dreambooth/datasets/3-lowpoly/office/model
-pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
-prior_loss_weight: 1.0
-push_to_hub: false
-resolution: 512
-revision: null
-sample_batch_size: 4
-scale_lr: false
-seed: null
-tokenizer_name: null
-train_batch_size: 1
-train_text_encoder: true
-use_8bit_adam: true
-with_prior_preservation: false

logs/dreambooth/1669596008.5475128/events.out.tfevents.1669596008.instance-3.23115.1

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7ae54b9150bcb9b6a6929094f0008482fc1d2fa0495c7b0dbb44a6716a69ae07
-size 1908

logs/dreambooth/1669596008.5630267/hparams.yml

@@ -1,37 +0,0 @@
-adam_beta1: 0.9
-adam_beta2: 0.999
-adam_epsilon: 1.0e-08
-adam_weight_decay: 0.01
-center_crop: false
-class_data_dir: null
-class_prompt: null
-gradient_accumulation_steps: 1
-gradient_checkpointing: true
-hub_model_id: null
-hub_token: null
-instance_data_dir: /home/diffusers/examples/dreambooth/datasets/3-lowpoly/office/class
-instance_prompt: a photo of lowpoly_office
-learning_rate: 2.0e-06
-local_rank: -1
-logging_dir: logs
-lr_scheduler: constant
-lr_warmup_steps: 0
-max_grad_norm: 1.0
-max_train_steps: 2000
-mixed_precision: null
-num_class_images: 100
-num_train_epochs: 223
-output_dir: /home/diffusers/examples/dreambooth/datasets/3-lowpoly/office/model
-pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
-prior_loss_weight: 1.0
-push_to_hub: false
-resolution: 512
-revision: null
-sample_batch_size: 4
-scale_lr: false
-seed: null
-tokenizer_name: null
-train_batch_size: 1
-train_text_encoder: true
-use_8bit_adam: true
-with_prior_preservation: false

logs/dreambooth/1669596111.2761257/events.out.tfevents.1669596111.instance-3.24326.1

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:2b2a8816cd8ebdd1c77d3cc5e7d02b6997f5fa7b04f0554c8a9f5d3224711219
-size 1908

logs/dreambooth/1669596111.2919507/hparams.yml

@@ -1,37 +0,0 @@
-adam_beta1: 0.9
-adam_beta2: 0.999
-adam_epsilon: 1.0e-08
-adam_weight_decay: 0.01
-center_crop: false
-class_data_dir: null
-class_prompt: null
-gradient_accumulation_steps: 1
-gradient_checkpointing: true
-hub_model_id: null
-hub_token: null
-instance_data_dir: /home/diffusers/examples/dreambooth/datasets/3-lowpoly/office/class
-instance_prompt: a photo of lowpoly_office
-learning_rate: 2.0e-06
-local_rank: -1
-logging_dir: logs
-lr_scheduler: constant
-lr_warmup_steps: 0
-max_grad_norm: 1.0
-max_train_steps: 2000
-mixed_precision: null
-num_class_images: 100
-num_train_epochs: 223
-output_dir: /home/diffusers/examples/dreambooth/datasets/3-lowpoly/office/model
-pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
-prior_loss_weight: 1.0
-push_to_hub: false
-resolution: 512
-revision: null
-sample_batch_size: 4
-scale_lr: false
-seed: null
-tokenizer_name: null
-train_batch_size: 1
-train_text_encoder: true
-use_8bit_adam: true
-with_prior_preservation: false

logs/dreambooth/1669596158.271865/events.out.tfevents.1669596158.instance-3.24889.1

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:1dd31d23173b0798ef2b521375a2df9df4c007ee02cbb2fd167ff553f7bc35bc
-size 1908

logs/dreambooth/1669596158.2885222/hparams.yml

@@ -1,37 +0,0 @@
-adam_beta1: 0.9
-adam_beta2: 0.999
-adam_epsilon: 1.0e-08
-adam_weight_decay: 0.01
-center_crop: false
-class_data_dir: null
-class_prompt: null
-gradient_accumulation_steps: 1
-gradient_checkpointing: true
-hub_model_id: null
-hub_token: null
-instance_data_dir: /home/diffusers/examples/dreambooth/datasets/3-lowpoly/office/class
-instance_prompt: a photo of lowpoly_office
-learning_rate: 2.0e-06
-local_rank: -1
-logging_dir: logs
-lr_scheduler: constant
-lr_warmup_steps: 0
-max_grad_norm: 1.0
-max_train_steps: 2000
-mixed_precision: null
-num_class_images: 100
-num_train_epochs: 223
-output_dir: /home/diffusers/examples/dreambooth/datasets/3-lowpoly/office/model
-pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
-prior_loss_weight: 1.0
-push_to_hub: false
-resolution: 512
-revision: null
-sample_batch_size: 4
-scale_lr: false
-seed: null
-tokenizer_name: null
-train_batch_size: 1
-train_text_encoder: true
-use_8bit_adam: true
-with_prior_preservation: false

logs/dreambooth/1669597391.7762234/events.out.tfevents.1669597391.instance-3.14316.1

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:36c44893e79bd5230f7cd7a8aa64053c266c0b7eec823ff18b3cb87c899d49e3
-size 1908

logs/dreambooth/1669597391.7921813/hparams.yml

@@ -1,37 +0,0 @@
-adam_beta1: 0.9
-adam_beta2: 0.999
-adam_epsilon: 1.0e-08
-adam_weight_decay: 0.01
-center_crop: false
-class_data_dir: null
-class_prompt: null
-gradient_accumulation_steps: 1
-gradient_checkpointing: true
-hub_model_id: null
-hub_token: null
-instance_data_dir: /home/diffusers/examples/dreambooth/datasets/3-lowpoly/office/class
-instance_prompt: a photo of lowpoly_office
-learning_rate: 2.0e-06
-local_rank: -1
-logging_dir: logs
-lr_scheduler: constant
-lr_warmup_steps: 0
-max_grad_norm: 1.0
-max_train_steps: 2000
-mixed_precision: null
-num_class_images: 100
-num_train_epochs: 223
-output_dir: /home/diffusers/examples/dreambooth/datasets/3-lowpoly/office/model
-pretrained_model_name_or_path: runwayml/stable-diffusion-v1-5
-prior_loss_weight: 1.0
-push_to_hub: false
-resolution: 512
-revision: null
-sample_batch_size: 4
-scale_lr: false
-seed: null
-tokenizer_name: null
-train_batch_size: 1
-train_text_encoder: true
-use_8bit_adam: true
-with_prior_preservation: false

logs/dreambooth/events.out.tfevents.1669594587.instance-3.1963.0

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:9dfdffdbf4973c0202fe89e6836c6717e06cead6ea4ed1d6ce08628fb8e30c6e
-size 167786

logs/dreambooth/events.out.tfevents.1669596008.instance-3.23115.0

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:54a9aee086fc63232e221726af4c529cee0620a609d2d9a979d5a56b8aa6e1a1
-size 942

logs/dreambooth/events.out.tfevents.1669596111.instance-3.24326.0

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:525eeb3285db746bf152663a59bc9c2ef07d5c315fd436780ee69ea298167260
-size 40

logs/dreambooth/events.out.tfevents.1669596158.instance-3.24889.0

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:dd4025e063e684e42a9597aee874d1d717ce8ef467af784c0334dec94e36eeae
-size 3812

logs/dreambooth/events.out.tfevents.1669597391.instance-3.14316.0

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:bc3edbaf4bc5f19ee4c4cf14bf598a04ef22af35e9b43d664a9f2ae7398c2ce1
-size 167786

model_index.json

@@ -2,13 +2,13 @@
   "_class_name": "StableDiffusionPipeline",
   "_diffusers_version": "0.9.0",
   "feature_extractor": [
-    "transformers",
-    "CLIPFeatureExtractor"
+    null,
+    null
   ],
-  "requires_safety_checker": true,
+  "requires_safety_checker": false,
   "safety_checker": [
-    "stable_diffusion",
-    "StableDiffusionSafetyChecker"
+    null,
+    null
   ],
   "scheduler": [
     "diffusers",

safety_checker/config.json

@@ -1,179 +0,0 @@
-{
-  "_commit_hash": "63534535d4730d5976c5c647a7f2adaea1102f5b",
-  "_name_or_path": "/root/.cache/huggingface/diffusers/models--runwayml--stable-diffusion-v1-5/snapshots/63534535d4730d5976c5c647a7f2adaea1102f5b/safety_checker",
-  "architectures": [
-    "StableDiffusionSafetyChecker"
-  ],
-  "initializer_factor": 1.0,
-  "logit_scale_init_value": 2.6592,
-  "model_type": "clip",
-  "projection_dim": 768,
-  "text_config": {
-    "_name_or_path": "",
-    "add_cross_attention": false,
-    "architectures": null,
-    "attention_dropout": 0.0,
-    "bad_words_ids": null,
-    "begin_suppress_tokens": null,
-    "bos_token_id": 0,
-    "chunk_size_feed_forward": 0,
-    "cross_attention_hidden_size": null,
-    "decoder_start_token_id": null,
-    "diversity_penalty": 0.0,
-    "do_sample": false,
-    "dropout": 0.0,
-    "early_stopping": false,
-    "encoder_no_repeat_ngram_size": 0,
-    "eos_token_id": 2,
-    "exponential_decay_length_penalty": null,
-    "finetuning_task": null,
-    "forced_bos_token_id": null,
-    "forced_eos_token_id": null,
-    "hidden_act": "quick_gelu",
-    "hidden_size": 768,
-    "id2label": {
-      "0": "LABEL_0",
-      "1": "LABEL_1"
-    },
-    "initializer_factor": 1.0,
-    "initializer_range": 0.02,
-    "intermediate_size": 3072,
-    "is_decoder": false,
-    "is_encoder_decoder": false,
-    "label2id": {
-      "LABEL_0": 0,
-      "LABEL_1": 1
-    },
-    "layer_norm_eps": 1e-05,
-    "length_penalty": 1.0,
-    "max_length": 20,
-    "max_position_embeddings": 77,
-    "min_length": 0,
-    "model_type": "clip_text_model",
-    "no_repeat_ngram_size": 0,
-    "num_attention_heads": 12,
-    "num_beam_groups": 1,
-    "num_beams": 1,
-    "num_hidden_layers": 12,
-    "num_return_sequences": 1,
-    "output_attentions": false,
-    "output_hidden_states": false,
-    "output_scores": false,
-    "pad_token_id": 1,
-    "prefix": null,
-    "problem_type": null,
-    "pruned_heads": {},
-    "remove_invalid_values": false,
-    "repetition_penalty": 1.0,
-    "return_dict": true,
-    "return_dict_in_generate": false,
-    "sep_token_id": null,
-    "suppress_tokens": null,
-    "task_specific_params": null,
-    "temperature": 1.0,
-    "tf_legacy_loss": false,
-    "tie_encoder_decoder": false,
-    "tie_word_embeddings": true,
-    "tokenizer_class": null,
-    "top_k": 50,
-    "top_p": 1.0,
-    "torch_dtype": null,
-    "torchscript": false,
-    "transformers_version": "4.24.0",
-    "typical_p": 1.0,
-    "use_bfloat16": false,
-    "vocab_size": 49408
-  },
-  "text_config_dict": {
-    "hidden_size": 768,
-    "intermediate_size": 3072,
-    "num_attention_heads": 12,
-    "num_hidden_layers": 12
-  },
-  "torch_dtype": "float32",
-  "transformers_version": null,
-  "vision_config": {
-    "_name_or_path": "",
-    "add_cross_attention": false,
-    "architectures": null,
-    "attention_dropout": 0.0,
-    "bad_words_ids": null,
-    "begin_suppress_tokens": null,
-    "bos_token_id": null,
-    "chunk_size_feed_forward": 0,
-    "cross_attention_hidden_size": null,
-    "decoder_start_token_id": null,
-    "diversity_penalty": 0.0,
-    "do_sample": false,
-    "dropout": 0.0,
-    "early_stopping": false,
-    "encoder_no_repeat_ngram_size": 0,
-    "eos_token_id": null,
-    "exponential_decay_length_penalty": null,
-    "finetuning_task": null,
-    "forced_bos_token_id": null,
-    "forced_eos_token_id": null,
-    "hidden_act": "quick_gelu",
-    "hidden_size": 1024,
-    "id2label": {
-      "0": "LABEL_0",
-      "1": "LABEL_1"
-    },
-    "image_size": 224,
-    "initializer_factor": 1.0,
-    "initializer_range": 0.02,
-    "intermediate_size": 4096,
-    "is_decoder": false,
-    "is_encoder_decoder": false,
-    "label2id": {
-      "LABEL_0": 0,
-      "LABEL_1": 1
-    },
-    "layer_norm_eps": 1e-05,
-    "length_penalty": 1.0,
-    "max_length": 20,
-    "min_length": 0,
-    "model_type": "clip_vision_model",
-    "no_repeat_ngram_size": 0,
-    "num_attention_heads": 16,
-    "num_beam_groups": 1,
-    "num_beams": 1,
-    "num_channels": 3,
-    "num_hidden_layers": 24,
-    "num_return_sequences": 1,
-    "output_attentions": false,
-    "output_hidden_states": false,
-    "output_scores": false,
-    "pad_token_id": null,
-    "patch_size": 14,
-    "prefix": null,
-    "problem_type": null,
-    "pruned_heads": {},
-    "remove_invalid_values": false,
-    "repetition_penalty": 1.0,
-    "return_dict": true,
-    "return_dict_in_generate": false,
-    "sep_token_id": null,
-    "suppress_tokens": null,
-    "task_specific_params": null,
-    "temperature": 1.0,
-    "tf_legacy_loss": false,
-    "tie_encoder_decoder": false,
-    "tie_word_embeddings": true,
-    "tokenizer_class": null,
-    "top_k": 50,
-    "top_p": 1.0,
-    "torch_dtype": null,
-    "torchscript": false,
-    "transformers_version": "4.24.0",
-    "typical_p": 1.0,
-    "use_bfloat16": false
-  },
-  "vision_config_dict": {
-    "hidden_size": 1024,
-    "intermediate_size": 4096,
-    "num_attention_heads": 16,
-    "num_hidden_layers": 24,
-    "patch_size": 14
-  }
-}

safety_checker/pytorch_model.bin

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:16d28f2b37109f222cdc33620fdd262102ac32112be0352a7f77e9614b35a394
-size 1216064769

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "runwayml/stable-diffusion-v1-5",
+  "_name_or_path": "stabilityai/stable-diffusion-2-base",
   "architectures": [
     "CLIPTextModel"
   ],
@@ -7,18 +7,18 @@
   "bos_token_id": 0,
   "dropout": 0.0,
   "eos_token_id": 2,
-  "hidden_act": "quick_gelu",
-  "hidden_size": 768,
+  "hidden_act": "gelu",
+  "hidden_size": 1024,
   "initializer_factor": 1.0,
   "initializer_range": 0.02,
-  "intermediate_size": 3072,
+  "intermediate_size": 4096,
   "layer_norm_eps": 1e-05,
   "max_position_embeddings": 77,
   "model_type": "clip_text_model",
-  "num_attention_heads": 12,
-  "num_hidden_layers": 12,
+  "num_attention_heads": 16,
+  "num_hidden_layers": 23,
   "pad_token_id": 1,
-  "projection_dim": 768,
+  "projection_dim": 512,
   "torch_dtype": "float32",
   "transformers_version": "4.24.0",
   "vocab_size": 49408

text_encoder/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1791c589b8ca7805b231a9909fb6f8f61b4c8f117b40794746c2599951f9341f
-size 492309793
+oid sha256:47763632e668151e91062efebb4378a5569010abdd02c036cecac10621b68f84
+size 1361679905

tokenizer/special_tokens_map.json

@@ -13,7 +13,7 @@
     "rstrip": false,
     "single_word": false
   },
-  "pad_token": "<|endoftext|>",
+  "pad_token": "!",
   "unk_token": {
     "content": "<|endoftext|>",
     "lstrip": false,

tokenizer/tokenizer_config.json

@@ -19,7 +19,7 @@
   },
   "errors": "replace",
   "model_max_length": 77,
-  "name_or_path": "/root/.cache/huggingface/diffusers/models--runwayml--stable-diffusion-v1-5/snapshots/63534535d4730d5976c5c647a7f2adaea1102f5b/tokenizer",
+  "name_or_path": "/root/.cache/huggingface/diffusers/models--stabilityai--stable-diffusion-2-base/snapshots/abda38f7c548237cae935d6182ae56acd5a02ed8/tokenizer",
   "pad_token": "<|endoftext|>",
   "special_tokens_map_file": "./special_tokens_map.json",
   "tokenizer_class": "CLIPTokenizer",

unet/config.json

@@ -1,9 +1,14 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.9.0",
-  "_name_or_path": "runwayml/stable-diffusion-v1-5",
+  "_name_or_path": "stabilityai/stable-diffusion-2-base",
   "act_fn": "silu",
-  "attention_head_dim": 8,
+  "attention_head_dim": [
+    5,
+    10,
+    20,
+    20
+  ],
   "block_out_channels": [
     320,
     640,
@@ -11,7 +16,7 @@
     1280
   ],
   "center_input_sample": false,
-  "cross_attention_dim": 768,
+  "cross_attention_dim": 1024,
   "down_block_types": [
     "CrossAttnDownBlock2D",
     "CrossAttnDownBlock2D",
@@ -37,5 +42,5 @@
     "CrossAttnUpBlock2D",
     "CrossAttnUpBlock2D"
   ],
-  "use_linear_projection": false
+  "use_linear_projection": true
 }

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d616a67487028c6c96781aaa846d248a311aaf7c9f9ecdb6713e0ac58a203b81
-size 3438375973
+oid sha256:abe310d87b0521d781563ff62b3168b93c130c7cf923daab3ede4f8dbc37c607
+size 3463934693

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.9.0",
-  "_name_or_path": "/root/.cache/huggingface/diffusers/models--runwayml--stable-diffusion-v1-5/snapshots/63534535d4730d5976c5c647a7f2adaea1102f5b/vae",
+  "_name_or_path": "/root/.cache/huggingface/diffusers/models--stabilityai--stable-diffusion-2-base/snapshots/abda38f7c548237cae935d6182ae56acd5a02ed8/vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,

vae/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:af27ea858349760ebe3311953e0bfe8d6fd257dc9537ae0b2b938c262132a2c6
+oid sha256:36bb8e1b54aba3a0914eb35fba13dcb107e9f18d379d1df2158732cd4bf56a94
 size 334711857