epoch 1

feature_extractor/preprocessor_config.json

@@ -0,0 +1,28 @@
+{
+  "crop_size": {
+    "height": 224,
+    "width": 224
+  },
+  "do_center_crop": true,
+  "do_convert_rgb": true,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "feature_extractor_type": "CLIPFeatureExtractor",
+  "image_mean": [
+    0.48145466,
+    0.4578275,
+    0.40821073
+  ],
+  "image_processor_type": "CLIPImageProcessor",
+  "image_std": [
+    0.26862954,
+    0.26130258,
+    0.27577711
+  ],
+  "resample": 3,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "shortest_edge": 224
+  }
+}

model_index.json

@@ -0,0 +1,32 @@
+{
+  "_class_name": "FlaxStableDiffusionPipeline",
+  "_diffusers_version": "0.12.0.dev0",
+  "feature_extractor": [
+    "transformers",
+    "CLIPImageProcessor"
+  ],
+  "safety_checker": [
+    null,
+    null
+  ],
+  "scheduler": [
+    "diffusers",
+    "FlaxPNDMScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "FlaxCLIPTextModel"
+  ],
+  "tokenizer": [
+    "transformers",
+    "CLIPTokenizer"
+  ],
+  "unet": [
+    "diffusers",
+    "FlaxUNet2DConditionModel"
+  ],
+  "vae": [
+    "diffusers",
+    "FlaxAutoencoderKL"
+  ]
+}

scheduler/scheduler_config.json

@@ -0,0 +1,14 @@
+{
+  "_class_name": "FlaxPNDMScheduler",
+  "_diffusers_version": "0.12.0.dev0",
+  "beta_end": 0.012,
+  "beta_schedule": "scaled_linear",
+  "beta_start": 0.00085,
+  "clip_sample": false,
+  "num_train_timesteps": 1000,
+  "prediction_type": "epsilon",
+  "set_alpha_to_one": false,
+  "skip_prk_steps": true,
+  "steps_offset": 1,
+  "trained_betas": null
+}

text_encoder/config.json

@@ -0,0 +1,25 @@
+{
+  "_name_or_path": "runwayml/stable-diffusion-v1-5",
+  "architectures": [
+    "CLIPTextModel"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 0,
+  "dropout": 0.0,
+  "eos_token_id": 2,
+  "hidden_act": "quick_gelu",
+  "hidden_size": 768,
+  "initializer_factor": 1.0,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "max_position_embeddings": 77,
+  "model_type": "clip_text_model",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 12,
+  "pad_token_id": 1,
+  "projection_dim": 768,
+  "torch_dtype": "float32",
+  "transformers_version": "4.25.1",
+  "vocab_size": 49501
+}

text_encoder/flax_model.msgpack

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

tokenizer/added_tokens.json

@@ -0,0 +1,95 @@
+{
+  "2_penises": 49484,
+  "3d_artwork": 49467,
+  "69_position": 49491,
+  "after_anal": 49480,
+  "all_fours": 49452,
+  "anal_fingering": 49482,
+  "anal_knotting": 49481,
+  "anal_penetration": 49413,
+  "anthro_on_anthro": 49420,
+  "anthro_penetrated": 49426,
+  "anthro_penetrating": 49425,
+  "anthro_penetrating_anthro": 49429,
+  "ass_up": 49449,
+  "autofellatio": 49485,
+  "barely_visible_genitalia": 49476,
+  "bent_over": 49444,
+  "buttplug": 49470,
+  "chair_position": 49483,
+  "circumcised": 49472,
+  "close-up": 49456,
+  "cowgirl_position": 49493,
+  "cum_inside": 49415,
+  "cum_on_penis": 49440,
+  "dildo_in_ass": 49474,
+  "dildo_sitting": 49469,
+  "double_penetration": 49458,
+  "duo_focus": 49462,
+  "explicit_content": 49499,
+  "female/female": 49418,
+  "female_on_top": 49496,
+  "female_penetrated": 49422,
+  "female_penetrating": 49412,
+  "feral_penetrated": 49451,
+  "feral_penetrating": 49450,
+  "fox_mccloud": 49447,
+  "from_behind_position": 49434,
+  "from_front_position": 49437,
+  "fully_clothed": 49488,
+  "gangbang": 49475,
+  "goodaes": 49410,
+  "greataes": 49411,
+  "group_sex": 49500,
+  "gynomorph": 49433,
+  "gynomorph_penetrated": 49490,
+  "gynomorph_penetrating": 49459,
+  "hand_on_butt": 49439,
+  "hands_behind_head": 49494,
+  "hellhound": 49497,
+  "hot_dogging": 49489,
+  "human_on_anthro": 49445,
+  "inner_ear_fluff": 49436,
+  "judy_hopps": 49471,
+  "kobold": 49465,
+  "legoshi_beastars": 49486,
+  "legs_up": 49448,
+  "looking_at_another": 49438,
+  "looking_at_viewer": 49414,
+  "looking_pleasured": 49428,
+  "loona_helluva_boss": 49453,
+  "lucario": 49409,
+  "male/female": 49417,
+  "male/male": 49416,
+  "male_on_top": 49455,
+  "male_penetrated": 49424,
+  "male_penetrating": 49408,
+  "mating_press": 49487,
+  "missionary_position": 49492,
+  "multi_genitalia": 49479,
+  "my_little_pony": 49446,
+  "older_male": 49457,
+  "partially_clothed": 49435,
+  "penetrating_pov": 49477,
+  "penile_penetration": 49421,
+  "penis_in_ass": 49443,
+  "pink_penis": 49442,
+  "pokephilia": 49498,
+  "predator/prey": 49478,
+  "presenting_anus": 49441,
+  "presenting_hindquarters": 49430,
+  "raccoon": 49464,
+  "raised_tail": 49431,
+  "rear_view": 49432,
+  "red_penis": 49460,
+  "size_difference": 49427,
+  "smirk": 49466,
+  "spread_anus": 49461,
+  "spread_legs": 49419,
+  "standing_sex": 49495,
+  "star_fox": 49463,
+  "thick_thighs": 49423,
+  "uncircumcised": 49473,
+  "wolf_odonnel": 49468,
+  "zootopia": 49454
+}

tokenizer/special_tokens_map.json

@@ -0,0 +1,24 @@
+{
+  "bos_token": {
+    "content": "<|startoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<|endoftext|>",
+  "unk_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer/tokenizer_config.json

@@ -0,0 +1,34 @@
+{
+  "add_prefix_space": false,
+  "bos_token": {
+    "__type": "AddedToken",
+    "content": "<|startoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "do_lower_case": true,
+  "eos_token": {
+    "__type": "AddedToken",
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "errors": "replace",
+  "model_max_length": 77,
+  "name_or_path": "/root/stable-diffusion-v1-5",
+  "pad_token": "<|endoftext|>",
+  "special_tokens_map_file": "./special_tokens_map.json",
+  "tokenizer_class": "CLIPTokenizer",
+  "unk_token": {
+    "__type": "AddedToken",
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
+}

unet/config.json

@@ -0,0 +1,39 @@
+{
+  "_class_name": "FlaxUNet2DConditionModel",
+  "_diffusers_version": "0.12.0.dev0",
+  "act_fn": "silu",
+  "attention_head_dim": 8,
+  "block_out_channels": [
+    320,
+    640,
+    1280,
+    1280
+  ],
+  "center_input_sample": false,
+  "cross_attention_dim": 768,
+  "down_block_types": [
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "CrossAttnDownBlock2D",
+    "DownBlock2D"
+  ],
+  "downsample_padding": 1,
+  "dropout": 0.0,
+  "flip_sin_to_cos": true,
+  "freq_shift": 0,
+  "in_channels": 4,
+  "layers_per_block": 2,
+  "mid_block_scale_factor": 1,
+  "norm_eps": 1e-05,
+  "norm_num_groups": 32,
+  "only_cross_attention": false,
+  "out_channels": 4,
+  "sample_size": 64,
+  "up_block_types": [
+    "UpBlock2D",
+    "CrossAttnUpBlock2D",
+    "CrossAttnUpBlock2D",
+    "CrossAttnUpBlock2D"
+  ],
+  "use_linear_projection": false
+}

unet/diffusion_flax_model.msgpack

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

vae/config.json

@@ -0,0 +1,29 @@
+{
+  "_class_name": "FlaxAutoencoderKL",
+  "_diffusers_version": "0.12.0.dev0",
+  "act_fn": "silu",
+  "block_out_channels": [
+    128,
+    256,
+    512,
+    512
+  ],
+  "down_block_types": [
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D"
+  ],
+  "in_channels": 3,
+  "latent_channels": 4,
+  "layers_per_block": 2,
+  "norm_num_groups": 32,
+  "out_channels": 3,
+  "sample_size": 512,
+  "up_block_types": [
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D"
+  ]
+}

vae/diffusion_flax_model.msgpack

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