End of training

README.md

@@ -1,7 +1,7 @@
 
 ---
 license: creativeml-openrail-m
-base_model: runwayml/stable-diffusion-v1-5
+base_model: stabilityai/stable-diffusion-2-1
 tags:
 - stable-diffusion
 - stable-diffusion-diffusers
@@ -12,6 +12,6 @@ inference: true
 ---
     
 # Textual inversion text2image fine-tuning - thanhnguyentung/textual_inversion_cat
-These are textual inversion adaption weights for runwayml/stable-diffusion-v1-5. You can find some example images in the following. 
+These are textual inversion adaption weights for stabilityai/stable-diffusion-2-1. You can find some example images in the following. 
 
 

checkpoint-1000/optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a2d68976782a3aec8c12952ad67f7c5d9ff5b08727ba34c9e62031419150d04f
-size 303570477
+oid sha256:a67c65dfdc8a47371b3737de257cf174413fc8f013cd8507b3897c383710e558
+size 404760109

checkpoint-1000/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bfe757d8717f4a5ea864569cc3dc6e154bda39fc6c9616f5259e599a5c4105ca
-size 492322337
+oid sha256:d0e3df955417aab2eacb3c00c1f582cae3b0555b34ad96173c04093098c0d61b
+size 1361701921

checkpoint-1000/random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:80bb4c3ab5907474487e42d422edd8b5bf8b5d608c0d64b4035fd10f1b522611
+oid sha256:6c2cd3a3712ddcc84b4fe7449e27e5583e6beb08077af25d5f374c55ace03049
 size 15691

checkpoint-1000/scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:715fa16b4fd4791aa0a984f5fb08045ac05c6514a57c227259bdf620d6626770
+oid sha256:bbb19b948b33a080098712534e53a991114e50367c7efbe3f9b4d0cec03eb540
 size 563

checkpoint-1500/optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:473f80953dda42488d9da5aa4525bebf9e475f5fa0f999407e70b0405ea4d7f4
-size 303570477
+oid sha256:c0f5b082250174d7ef7dddf45fa7ff322c46a73175ff4d356843727c7681dca4
+size 404760109

checkpoint-1500/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5ec238ab51b68df6289eb87533c812d0068f468e5aa0128f593787bae3706c56
-size 492322337
+oid sha256:9dfb915b76de1b7d14eb301fbb39e2cd3e340aa3a5b4298f98c5ece82b98bb2b
+size 1361701921

checkpoint-1500/random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:94f012119de6ecc00ffb7d44d9c7b551421c8bbccdaf358c51dabc22d3b22275
-size 15627
+oid sha256:45b81fdac6e0a3ac260ba2ec496f6bd212111c5a4f6ad59f66b4adc04abc003b
+size 15691

checkpoint-1500/scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9c9c53bae75902cf0bda4ca44182b1fbddaaf7133c960eff4abda69a0b939b24
+oid sha256:7abf0b3af2013b920618166f35e17d899fce776df72f2c2a64c13989f4b8a6a8
 size 563

checkpoint-2000/optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a9daf9f757fa9ffcaa6517c17b29283915c94f00df740c5edb220b33c48da40b
-size 303570477
+oid sha256:f28cc66ccab078b8a1dbcd32512dc40930ed1a39d0a2bd4612ba733839a98488
+size 404760109

checkpoint-2000/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5acff81585df78824aa494c02d3120e858afc976774d69ad0190699ae79c678b
-size 492322337
+oid sha256:3ec93c3b287d359188976caac6fadc6dafe77900ec5dc0beb5694033971e0a61
+size 1361701921

checkpoint-2000/random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ac2fc0cf66a26959ec242710d1b81829379f0e5147551e074d04d51dcafdc43f
-size 15691
+oid sha256:f3bfb8a8c7eac15b0cba851f989520f4b67f9ce59f7e885b433dc7e561963970
+size 15627

checkpoint-2000/scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:daeee60414f0809b07c82047b035fbe1bcb76d06ad64f326328c473ee9ffa299
+oid sha256:7ce55ada40659c32ea01f44d7732f0406a6c9004376f7e5579de396b3481cc2d
 size 563

checkpoint-2500/optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a1fc6e8bdfa8559902ff06bbe748c767f5fed0436f3782e81faed999ed5ff162
-size 303570477
+oid sha256:f240bc838b091efc35a1b2f6ed6a7d08736f1b3eb6a8875695c60a3547c1731e
+size 404760109

checkpoint-2500/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f7eeff071d0f5947659acf14a91df96c18ea3ceb263e5aeb75e073a8f5f313c6
-size 492322337
+oid sha256:416bf2e0acaf2efcf20c3705ffcfc72c2a980accf757d216684342bbf57f794f
+size 1361701921

checkpoint-2500/random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:766139c0cf9719e5d8df6a0e12794ac77e01433e86bd2bebc2811ec86a2bab3f
-size 15691
+oid sha256:fd7def25a6148289db6b026b797379132a820ef9b86e5c72dc964f8e7143903b
+size 15755

checkpoint-2500/scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:86d6d2ca7543df568eb16c4516c191ee12db67c77fc9ec97e59cdaac4bd29460
+oid sha256:496e5a35d8686f4bf9ce59c376f5f501ec4a1034cf3b70deb1f3d2c047fab2d4
 size 563

checkpoint-3000/optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9213a9b2659d826356448cf2dce461d6563cc48758a1d3f14330526b1e6e0e5b
-size 303570477
+oid sha256:0cc149728c2118b7b3a5f84eba181057c06a2808403ec055956f9c30456372c8
+size 404760109

checkpoint-3000/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3623b5f122ffc7b8283ead5528434ab06cb4752013ad4ae43d8ab188e0914084
-size 492322337
+oid sha256:e94b3c396bbb4da18faf0ff7f91741eaadfd8a162c12e5c8b4d80842e63b8e78
+size 1361701921

checkpoint-3000/random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:03a6dd7a418a2208db7a24d083d1cf43967b3f87d5328fcbce3f13a98d4031ac
+oid sha256:71dc14af90465f31092e99074e99f6772371c94875ae7666bbd1d5a500a8acb5
 size 15691

checkpoint-3000/scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4bb67b14cf0223dfbb9873a094fa17d5b2c07a172b15c32f302d03b6f8e1e9ff
+oid sha256:2056b541ceef5d77999a48951b85854f6503b83d58bc070e970ec3531e56e169
 size 563

checkpoint-500/optimizer.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:31cd2aa3308c4c3a2800d7b449637ce3a6a47ec500396f5d61081b8d9b361b39
-size 303570477
+oid sha256:bb988b7441972a13e4e6a03b0113f1861d21343ea44a57eaf4824db52466d823
+size 404760109

checkpoint-500/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b8ee0f3c7cbe30025f3257b6b3083abe9b6eb09b629aa1cf85f81e3b3ba218dd
-size 492322337
+oid sha256:d9a01433566f609f24383285b9c18fd9ee5e54b183788f4970da6189b9a0f46f
+size 1361701921

checkpoint-500/random_states_0.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ef953919d2cf445bf47815e4b58df78a7052ef10f5e43d39d8aef9a0e5d1a79e
-size 15627
+oid sha256:f7ca7c26b7c91bd4fa127d788ccb981969f2a5a380ef5dd1778dca15d9589043
+size 15691

checkpoint-500/scheduler.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:74481ac148859637e49c5946a58db09e35f78ebf008612a8ee0b58a9f39e1844
+oid sha256:478ff7b9adcb1fba8f74abd29e0dfb41133888ecf608d07ea8ae2143e019d098
 size 563

learned_embeds-steps-1000.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a0d795ff545f9f071752ed7358c5c7abf9570303227042b071e76bcb46a628b8
-size 4001
+oid sha256:68457e561e4805acd9f380c5ea739e9ed1b2804ffe65a944f229745c74e64d60
+size 5025

learned_embeds-steps-1500.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d34a6651309047bc03f51380dc259313d74f527a54e5f61e25b5d2318db70eef
-size 4001
+oid sha256:82099d853a76a8c5062fd99a520a8377be9c1a5b2345920301d52e2e1d915ac0
+size 5025

learned_embeds-steps-2000.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8ad6c4e0ed2ba8fcb76d7a02ad79417ddf1c877bbad4feb1687ae0ab00870eb6
-size 4001
+oid sha256:78c4ccfcf824f47e48bc26a45877a902fa711580129cf2585c30e13736628c70
+size 5025

learned_embeds-steps-2500.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0d62a3352bc72b39362f392b7742edbc82a853b71d92ac3ebe1b372d0f1fe266
-size 4001
+oid sha256:f5173150b11e15283e14bce35105eaeee1537f08bc0e9980c0ce31cc0b86cd3b
+size 5025

learned_embeds-steps-3000.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:74d4fb7b64d4256d95e1dba83288169550b862c56d8281af4d3b3593b2d7e2dc
-size 4001
+oid sha256:4f64c6e5850f952a82c48fe66882e312b41b1c321c63f406650b10f6bd650166
+size 5025

learned_embeds-steps-500.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:aff8160124907cd8ed374209f2e66b0ab1b572725862ac9b27951a94a7f7326f
-size 3998
+oid sha256:4ba40796fd3cf6785f608a04e5bec6bbb0174f8068f03a4b1d5927bb81d6e022
+size 5022

learned_embeds.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c3b6c1fef41db9c32c7b6b26518b488c878a5589e2007a3af3cce1c04f9d44a6
-size 3840
+oid sha256:54eeac57d11a838eb5783f170748c102a77512be54ec1b971639b3e196c309aa
+size 4864

logs/textual_inversion/1687311936.238098/events.out.tfevents.1687311936.sun-rnd.2641011.1

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

logs/textual_inversion/1687311936.2410269/hparams.yml

@@ -0,0 +1,46 @@
+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: 500
+checkpoints_total_limit: null
+dataloader_num_workers: 0
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+hub_model_id: null
+hub_token: null
+initializer_token: man
+learnable_property: object
+learning_rate: 0.002
+local_rank: -1
+logging_dir: logs
+lr_num_cycles: 1
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_train_steps: 3000
+mixed_precision: 'no'
+num_train_epochs: 6
+num_validation_images: 4
+num_vectors: 1
+output_dir: textual_inversion_cat
+placeholder_token: <thanh-nguyen>
+pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
+push_to_hub: true
+repeats: 100
+report_to: tensorboard
+resolution: 768
+resume_from_checkpoint: null
+revision: null
+save_as_full_pipeline: false
+save_steps: 500
+scale_lr: true
+seed: null
+tokenizer_name: null
+train_batch_size: 1
+train_data_dir: ./datasets/myportraits1024
+validation_epochs: null
+validation_prompt: null
+validation_steps: 100

logs/textual_inversion/1687311986.9854565/events.out.tfevents.1687311986.sun-rnd.2641279.1

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

logs/textual_inversion/1687311986.987646/hparams.yml

@@ -0,0 +1,46 @@
+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: 500
+checkpoints_total_limit: null
+dataloader_num_workers: 0
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+hub_model_id: null
+hub_token: null
+initializer_token: man
+learnable_property: object
+learning_rate: 0.002
+local_rank: -1
+logging_dir: logs
+lr_num_cycles: 1
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_train_steps: 3000
+mixed_precision: 'no'
+num_train_epochs: 6
+num_validation_images: 4
+num_vectors: 1
+output_dir: textual_inversion_cat
+placeholder_token: <thanh-nguyen>
+pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
+push_to_hub: true
+repeats: 100
+report_to: tensorboard
+resolution: 768
+resume_from_checkpoint: null
+revision: null
+save_as_full_pipeline: false
+save_steps: 500
+scale_lr: true
+seed: null
+tokenizer_name: null
+train_batch_size: 1
+train_data_dir: ./datasets/myportraits1024
+validation_epochs: null
+validation_prompt: null
+validation_steps: 100

logs/textual_inversion/1687312214.902746/events.out.tfevents.1687312214.sun-rnd.2641973.1

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

logs/textual_inversion/1687312214.9046993/hparams.yml

@@ -0,0 +1,46 @@
+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: 500
+checkpoints_total_limit: null
+dataloader_num_workers: 0
+enable_xformers_memory_efficient_attention: false
+gradient_accumulation_steps: 4
+gradient_checkpointing: false
+hub_model_id: null
+hub_token: null
+initializer_token: man
+learnable_property: object
+learning_rate: 0.002
+local_rank: -1
+logging_dir: logs
+lr_num_cycles: 1
+lr_scheduler: constant
+lr_warmup_steps: 0
+max_train_steps: 3000
+mixed_precision: 'no'
+num_train_epochs: 6
+num_validation_images: 4
+num_vectors: 1
+output_dir: textual_inversion_cat
+placeholder_token: <thanh-nguyen>
+pretrained_model_name_or_path: stabilityai/stable-diffusion-2-1
+push_to_hub: true
+repeats: 100
+report_to: tensorboard
+resolution: 768
+resume_from_checkpoint: null
+revision: null
+save_as_full_pipeline: false
+save_steps: 500
+scale_lr: true
+seed: null
+tokenizer_name: null
+train_batch_size: 1
+train_data_dir: ./datasets/myportraits1024
+validation_epochs: null
+validation_prompt: null
+validation_steps: 100

logs/textual_inversion/events.out.tfevents.1687311936.sun-rnd.2641011.0

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

logs/textual_inversion/events.out.tfevents.1687311986.sun-rnd.2641279.0

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

logs/textual_inversion/events.out.tfevents.1687312214.sun-rnd.2641973.0

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

model_index.json

@@ -5,14 +5,14 @@
     "transformers",
     "CLIPImageProcessor"
   ],
-  "requires_safety_checker": true,
+  "requires_safety_checker": false,
   "safety_checker": [
-    "stable_diffusion",
-    "StableDiffusionSafetyChecker"
+    null,
+    null
   ],
   "scheduler": [
     "diffusers",
-    "PNDMScheduler"
+    "DDIMScheduler"
   ],
   "text_encoder": [
     "transformers",

scheduler/scheduler_config.json

@@ -1,14 +1,20 @@
 {
-  "_class_name": "PNDMScheduler",
+  "_class_name": "DDIMScheduler",
   "_diffusers_version": "0.18.0.dev0",
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,
   "clip_sample": false,
+  "clip_sample_range": 1.0,
+  "dynamic_thresholding_ratio": 0.995,
   "num_train_timesteps": 1000,
-  "prediction_type": "epsilon",
+  "prediction_type": "v_prediction",
+  "rescale_betas_zero_snr": false,
+  "sample_max_value": 1.0,
   "set_alpha_to_one": false,
   "skip_prk_steps": true,
   "steps_offset": 1,
+  "thresholding": false,
+  "timestep_spacing": "leading",
   "trained_betas": null
 }

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "runwayml/stable-diffusion-v1-5",
+  "_name_or_path": "stabilityai/stable-diffusion-2-1",
   "architectures": [
     "CLIPTextModel"
   ],
@@ -7,19 +7,19 @@
   "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.30.2",
+  "transformers_version": "4.26.1",
   "vocab_size": 49409
 }

text_encoder/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9f3214411794ea0828e3ee19710e591c45ee46af056393ad62668fc356ee1831
-size 492312865
+oid sha256:e2b70406163946c4d0404ffe65231e1c03afa03d80c2dbd6908548c0298bdde0
+size 1361684001

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

@@ -8,7 +8,6 @@
     "rstrip": false,
     "single_word": false
   },
-  "clean_up_tokenization_spaces": true,
   "do_lower_case": true,
   "eos_token": {
     "__type": "AddedToken",
@@ -20,7 +19,9 @@
   },
   "errors": "replace",
   "model_max_length": 77,
+  "name_or_path": "stabilityai/stable-diffusion-2-1",
   "pad_token": "<|endoftext|>",
+  "special_tokens_map_file": "./special_tokens_map.json",
   "tokenizer_class": "CLIPTokenizer",
   "unk_token": {
     "__type": "AddedToken",

unet/config.json

@@ -1,11 +1,16 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.18.0.dev0",
-  "_name_or_path": "runwayml/stable-diffusion-v1-5",
+  "_name_or_path": "stabilityai/stable-diffusion-2-1",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,
-  "attention_head_dim": 8,
+  "attention_head_dim": [
+    5,
+    10,
+    20,
+    20
+  ],
   "block_out_channels": [
     320,
     640,
@@ -17,7 +22,7 @@
   "class_embeddings_concat": false,
   "conv_in_kernel": 3,
   "conv_out_kernel": 3,
-  "cross_attention_dim": 768,
+  "cross_attention_dim": 1024,
   "cross_attention_norm": null,
   "down_block_types": [
     "CrossAttnDownBlock2D",
@@ -45,7 +50,7 @@
   "resnet_out_scale_factor": 1.0,
   "resnet_skip_time_act": false,
   "resnet_time_scale_shift": "default",
-  "sample_size": 64,
+  "sample_size": 96,
   "time_cond_proj_dim": null,
   "time_embedding_act_fn": null,
   "time_embedding_dim": null,
@@ -57,6 +62,6 @@
     "CrossAttnUpBlock2D",
     "CrossAttnUpBlock2D"
   ],
-  "upcast_attention": false,
-  "use_linear_projection": false
+  "upcast_attention": true,
+  "use_linear_projection": true
 }

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:39aeb5f57e936fdb52331f49d69299fb130a5456c269b67c08699b109e8485fd
-size 3438375973
+oid sha256:29c7f15709062e790deddc2c6d3d6a3e0a99d4a533b6bfd187bdea64a5e7a961
+size 3463934693

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.18.0.dev0",
-  "_name_or_path": "runwayml/stable-diffusion-v1-5",
+  "_name_or_path": "stabilityai/stable-diffusion-2-1",
   "act_fn": "silu",
   "block_out_channels": [
     128,
@@ -20,7 +20,7 @@
   "layers_per_block": 2,
   "norm_num_groups": 32,
   "out_channels": 3,
-  "sample_size": 512,
+  "sample_size": 768,
   "scaling_factor": 0.18215,
   "up_block_types": [
     "UpDecoderBlock2D",