.gitattributes CHANGED
@@ -1,9 +1,39 @@
1
- openjourney-v4.ckpt filter=lfs diff=lfs merge=lfs -text
2
- safety_checker/model.safetensors filter=lfs diff=lfs merge=lfs -text
3
- safety_checker/pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
4
- text_encoder/model.safetensors filter=lfs diff=lfs merge=lfs -text
5
- text_encoder/pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
6
- unet/diffusion_pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
7
- unet/diffusion_pytorch_model.safetensors filter=lfs diff=lfs merge=lfs -text
8
- vae/diffusion_pytorch_model.bin filter=lfs diff=lfs merge=lfs -text
9
- vae/diffusion_pytorch_model.safetensors filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tflite filter=lfs diff=lfs merge=lfs -text
29
+ *.tgz filter=lfs diff=lfs merge=lfs -text
30
+ *.wasm filter=lfs diff=lfs merge=lfs -text
31
+ *.xz filter=lfs diff=lfs merge=lfs -text
32
+ *.zip filter=lfs diff=lfs merge=lfs -text
33
+ *.zst filter=lfs diff=lfs merge=lfs -text
34
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
35
+ openjourney-epoch0 filter=lfs diff=lfs merge=lfs -text
36
+ openjourney-epoch1 filter=lfs diff=lfs merge=lfs -text
37
+ openjourney-epoch2 filter=lfs diff=lfs merge=lfs -text
38
+ openjourney-epoch3 filter=lfs diff=lfs merge=lfs -text
39
+ openjourney-epoch3-unpruned filter=lfs diff=lfs merge=lfs -text
README.md CHANGED
@@ -5,21 +5,11 @@ tags:
5
  - text-to-image
6
  pinned: true
7
  ---
 
8
 
9
- # <u>Openjourney v4</u>
10
- ## Trained on +124k Midjourney v4 images, by [PromptHero](https://prompthero.com/?utm_source=huggingface&utm_medium=referral)
11
 
12
- Trained on Stable Diffusion v1.5 using +124000 images, 12400 steps, 4 epochs +32 training hours.
13
 
14
- 💡 [Openjourney-v4 prompts](https://prompthero.com/openjourney-prompts?version=4)
15
-
16
-
17
- Pss... "mdjrny-v4 style" is not necessary anymore (yay!)
18
-
19
- 🎓 **Want to learn how to train Openjourney? 👉🏼 __[Join our course](https://prompthero.com/academy/dreambooth-stable-diffusion-train-fine-tune-course?utm_source=huggingface&utm_medium=referral)__ 🔥**
20
-
21
- <img src="https://s3.us-east-1.amazonaws.com/prompthero-newsletter/Group-66.png" alt="openjourney-v4" width="50%">
22
-
23
- # Openjourney Links
24
- - [Lora version](https://huggingface.co/prompthero/openjourney-lora)
25
- - [Openjourney Dreambooth](https://huggingface.co/prompthero/openjourney)
 
5
  - text-to-image
6
  pinned: true
7
  ---
8
+ This repo is for testing the first Openjourney fine tuned model.
9
 
10
+ It was trained over Stable Diffusion 1.5 with +60000 images, 4500 steps and 3 epochs.
 
11
 
12
+ So "mdjrny-v4 style" is not necessary anymore (yay!)
13
 
14
+ Things to check in this beta:
15
+ - Is the quality better that the [Dreambooth version](https://huggingface.co/prompthero/openjourney)?
 
 
 
 
 
 
 
 
 
 
model_index.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
- "_diffusers_version": "0.15.0.dev0",
4
  "feature_extractor": [
5
  "transformers",
6
- "CLIPImageProcessor"
7
  ],
8
  "requires_safety_checker": true,
9
  "safety_checker": [
 
1
  {
2
  "_class_name": "StableDiffusionPipeline",
3
+ "_diffusers_version": "0.11.0.dev0",
4
  "feature_extractor": [
5
  "transformers",
6
+ "CLIPFeatureExtractor"
7
  ],
8
  "requires_safety_checker": true,
9
  "safety_checker": [
safety_checker/model.safetensors → openjourney-v2-unpruned.ckpt RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9d6a233ff6fd5ccb9f76fd99618d73369c52dd3d8222376384d0e601911089e8
3
- size 1215981830
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f022143bce33d6fb645751e594665d25fcb00995b8475e4401acae2c1df471d4
3
+ size 12126929450
openjourney-v4.ckpt → openjourney-v2.ckpt RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:02e37aad9f74f574808ad456043b89e8c6b24e22828743fcf002168f76493d9b
3
- size 2132910209
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2fbe5ba6e2a842fa99f52500cd600ac33f510b6a76b58874fa92b39c72c5ad57
3
+ size 2132887840
safety_checker/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
- "_commit_hash": null,
3
- "_name_or_path": "./safety_checker",
4
  "architectures": [
5
  "StableDiffusionSafetyChecker"
6
  ],
@@ -80,11 +80,17 @@
80
  "top_p": 1.0,
81
  "torch_dtype": null,
82
  "torchscript": false,
83
- "transformers_version": "4.27.0.dev0",
84
  "typical_p": 1.0,
85
  "use_bfloat16": false,
86
  "vocab_size": 49408
87
  },
 
 
 
 
 
 
88
  "torch_dtype": "float32",
89
  "transformers_version": null,
90
  "vision_config": {
@@ -161,8 +167,15 @@
161
  "top_p": 1.0,
162
  "torch_dtype": null,
163
  "torchscript": false,
164
- "transformers_version": "4.27.0.dev0",
165
  "typical_p": 1.0,
166
  "use_bfloat16": false
 
 
 
 
 
 
 
167
  }
168
  }
 
1
  {
2
+ "_commit_hash": "cb41f3a270d63d454d385fc2e4f571c487c253c5",
3
+ "_name_or_path": "CompVis/stable-diffusion-safety-checker",
4
  "architectures": [
5
  "StableDiffusionSafetyChecker"
6
  ],
 
80
  "top_p": 1.0,
81
  "torch_dtype": null,
82
  "torchscript": false,
83
+ "transformers_version": "4.26.0.dev0",
84
  "typical_p": 1.0,
85
  "use_bfloat16": false,
86
  "vocab_size": 49408
87
  },
88
+ "text_config_dict": {
89
+ "hidden_size": 768,
90
+ "intermediate_size": 3072,
91
+ "num_attention_heads": 12,
92
+ "num_hidden_layers": 12
93
+ },
94
  "torch_dtype": "float32",
95
  "transformers_version": null,
96
  "vision_config": {
 
167
  "top_p": 1.0,
168
  "torch_dtype": null,
169
  "torchscript": false,
170
+ "transformers_version": "4.26.0.dev0",
171
  "typical_p": 1.0,
172
  "use_bfloat16": false
173
+ },
174
+ "vision_config_dict": {
175
+ "hidden_size": 1024,
176
+ "intermediate_size": 4096,
177
+ "num_attention_heads": 16,
178
+ "num_hidden_layers": 24,
179
+ "patch_size": 14
180
  }
181
  }
scheduler/scheduler_config.json CHANGED
@@ -1,18 +1,14 @@
1
  {
2
  "_class_name": "PNDMScheduler",
3
- "_diffusers_version": "0.15.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "clip_sample": false,
8
- "clip_sample_range": 1.0,
9
- "dynamic_thresholding_ratio": 0.995,
10
  "num_train_timesteps": 1000,
11
  "prediction_type": "epsilon",
12
- "sample_max_value": 1.0,
13
  "set_alpha_to_one": false,
14
  "skip_prk_steps": true,
15
  "steps_offset": 1,
16
- "thresholding": false,
17
  "trained_betas": null
18
  }
 
1
  {
2
  "_class_name": "PNDMScheduler",
3
+ "_diffusers_version": "0.11.0.dev0",
4
  "beta_end": 0.012,
5
  "beta_schedule": "scaled_linear",
6
  "beta_start": 0.00085,
7
  "clip_sample": false,
 
 
8
  "num_train_timesteps": 1000,
9
  "prediction_type": "epsilon",
 
10
  "set_alpha_to_one": false,
11
  "skip_prk_steps": true,
12
  "steps_offset": 1,
 
13
  "trained_betas": null
14
  }
text_encoder/config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "./text_encoder",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
@@ -20,6 +20,6 @@
20
  "pad_token_id": 1,
21
  "projection_dim": 768,
22
  "torch_dtype": "float32",
23
- "transformers_version": "4.27.0.dev0",
24
  "vocab_size": 49408
25
  }
 
1
  {
2
+ "_name_or_path": "openai/clip-vit-large-patch14",
3
  "architectures": [
4
  "CLIPTextModel"
5
  ],
 
20
  "pad_token_id": 1,
21
  "projection_dim": 768,
22
  "torch_dtype": "float32",
23
+ "transformers_version": "4.26.0.dev0",
24
  "vocab_size": 49408
25
  }
text_encoder/model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:94137adceaf0637d2001e12815408e6b3b4be8026110e7b3c7453b7c44db1ce6
3
- size 492265874
 
 
 
 
text_encoder/pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:10a7e470bf9dcee2628823009d05535c1e4187ec84bea9f3b85505a952306358
3
  size 492307041
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5356d97789073df587cc92ad80d82a11b011da72d1a4c054c60826426a35c53a
3
  size 492307041
tokenizer/tokenizer_config.json CHANGED
@@ -19,6 +19,7 @@
19
  },
20
  "errors": "replace",
21
  "model_max_length": 77,
 
22
  "pad_token": "<|endoftext|>",
23
  "special_tokens_map_file": "./special_tokens_map.json",
24
  "tokenizer_class": "CLIPTokenizer",
 
19
  },
20
  "errors": "replace",
21
  "model_max_length": 77,
22
+ "name_or_path": "openai/clip-vit-large-patch14",
23
  "pad_token": "<|endoftext|>",
24
  "special_tokens_map_file": "./special_tokens_map.json",
25
  "tokenizer_class": "CLIPTokenizer",
unet/config.json CHANGED
@@ -1,7 +1,6 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
- "_diffusers_version": "0.15.0.dev0",
4
- "_name_or_path": "./unet",
5
  "act_fn": "silu",
6
  "attention_head_dim": 8,
7
  "block_out_channels": [
@@ -11,9 +10,6 @@
11
  1280
12
  ],
13
  "center_input_sample": false,
14
- "class_embed_type": null,
15
- "conv_in_kernel": 3,
16
- "conv_out_kernel": 3,
17
  "cross_attention_dim": 768,
18
  "down_block_types": [
19
  "CrossAttnDownBlock2D",
@@ -28,18 +24,12 @@
28
  "in_channels": 4,
29
  "layers_per_block": 2,
30
  "mid_block_scale_factor": 1,
31
- "mid_block_type": "UNetMidBlock2DCrossAttn",
32
  "norm_eps": 1e-05,
33
  "norm_num_groups": 32,
34
  "num_class_embeds": null,
35
  "only_cross_attention": false,
36
  "out_channels": 4,
37
- "projection_class_embeddings_input_dim": null,
38
- "resnet_time_scale_shift": "default",
39
  "sample_size": 64,
40
- "time_cond_proj_dim": null,
41
- "time_embedding_type": "positional",
42
- "timestep_post_act": null,
43
  "up_block_types": [
44
  "UpBlock2D",
45
  "CrossAttnUpBlock2D",
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
+ "_diffusers_version": "0.11.0.dev0",
 
4
  "act_fn": "silu",
5
  "attention_head_dim": 8,
6
  "block_out_channels": [
 
10
  1280
11
  ],
12
  "center_input_sample": false,
 
 
 
13
  "cross_attention_dim": 768,
14
  "down_block_types": [
15
  "CrossAttnDownBlock2D",
 
24
  "in_channels": 4,
25
  "layers_per_block": 2,
26
  "mid_block_scale_factor": 1,
 
27
  "norm_eps": 1e-05,
28
  "norm_num_groups": 32,
29
  "num_class_embeds": null,
30
  "only_cross_attention": false,
31
  "out_channels": 4,
 
 
32
  "sample_size": 64,
 
 
 
33
  "up_block_types": [
34
  "UpBlock2D",
35
  "CrossAttnUpBlock2D",
unet/diffusion_pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:c95a09ddc504c0175cd5d24539ba45d154b02d47dc69f1682711c39034ef1fd0
3
  size 3438366373
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c3a308e8d2715f4c1869ffddea2ff61040b56e4181665002a78396b8e5f8c4d8
3
  size 3438366373
unet/diffusion_pytorch_model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:5e9d371c67da8911865895914f993775a06e3b5f3a2e721e8977abdf33d8cee7
3
- size 3438167540
 
 
 
 
vae/config.json CHANGED
@@ -1,7 +1,6 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
- "_diffusers_version": "0.15.0.dev0",
4
- "_name_or_path": "./vae",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
@@ -21,7 +20,6 @@
21
  "norm_num_groups": 32,
22
  "out_channels": 3,
23
  "sample_size": 512,
24
- "scaling_factor": 0.18215,
25
  "up_block_types": [
26
  "UpDecoderBlock2D",
27
  "UpDecoderBlock2D",
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
+ "_diffusers_version": "0.11.0.dev0",
 
4
  "act_fn": "silu",
5
  "block_out_channels": [
6
  128,
 
20
  "norm_num_groups": 32,
21
  "out_channels": 3,
22
  "sample_size": 512,
 
23
  "up_block_types": [
24
  "UpDecoderBlock2D",
25
  "UpDecoderBlock2D",
vae/diffusion_pytorch_model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:55ad9628530430eed28c01dcf3c971a29ffe5d657856cb054728b6c956da3037
3
- size 334643276