diff --git a/feature_extractor/preprocessor_config.json b/feature_extractor/preprocessor_config.json index 5294955ff7801083f720b34b55d0f1f51313c5c5..e0675735fd0052745f2f7b7291e4aa7859389998 100644 --- a/feature_extractor/preprocessor_config.json +++ b/feature_extractor/preprocessor_config.json @@ -1,8 +1,12 @@ { - "crop_size": 224, + "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": [ @@ -10,11 +14,15 @@ 0.4578275, 0.40821073 ], + "image_processor_type": "CLIPFeatureExtractor", "image_std": [ 0.26862954, 0.26130258, 0.27577711 ], "resample": 3, - "size": 224 + "rescale_factor": 0.00392156862745098, + "size": { + "shortest_edge": 224 + } } diff --git a/model_index.json b/model_index.json index 60a2364f4a55fe2efcd5a3a6cbff8f2791b5c4e3..a66e62c1051b40a0032911795a908a50accae65f 100644 --- a/model_index.json +++ b/model_index.json @@ -1,21 +1,22 @@ { "_class_name": "StableDiffusionPipeline", - "_diffusers_version": "0.4.0.dev0", + "_diffusers_version": "0.15.0.dev0", "feature_extractor": [ "transformers", "CLIPFeatureExtractor" ], + "requires_safety_checker": true, "safety_checker": [ "stable_diffusion", - "FlaxStableDiffusionSafetyChecker" + "StableDiffusionSafetyChecker" ], "scheduler": [ "diffusers", - "FlaxDDIMScheduler" + "DDIMScheduler" ], "text_encoder": [ "transformers", - "FlaxCLIPTextModel" + "CLIPTextModel" ], "tokenizer": [ "transformers", @@ -23,10 +24,10 @@ ], "unet": [ "diffusers", - "FlaxUNet2DConditionModel" + "UNet2DConditionModel" ], "vae": [ "diffusers", - "FlaxAutoencoderKL" + "AutoencoderKL" ] } diff --git a/safety_checker/config.json b/safety_checker/config.json index fb04a6b53c71cb8e9b59694e82ed2f04bcfccbce..e44fbb034712a40652458a1e15411231f9014cdc 100644 --- a/safety_checker/config.json +++ b/safety_checker/config.json @@ -1,6 +1,6 @@ { - "_commit_hash": "04b741590ff4e18ef4d778c72afe3ca3a680a0c7", - "_name_or_path": "hf-internal-testing/tiny-random-clip", + "_commit_hash": null, + "_name_or_path": "./safety_checker", "architectures": [ "StableDiffusionSafetyChecker" ], @@ -14,6 +14,7 @@ "architectures": null, "attention_dropout": 0.1, "bad_words_ids": null, + "begin_suppress_tokens": null, "bos_token_id": 0, "chunk_size_feed_forward": 0, "cross_attention_hidden_size": null, @@ -62,12 +63,14 @@ "pad_token_id": 1, "prefix": null, "problem_type": null, + "projection_dim": 512, "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, @@ -78,82 +81,11 @@ "top_p": 1.0, "torch_dtype": null, "torchscript": false, - "transformers_version": "4.22.0.dev0", + "transformers_version": "4.28.0.dev0", "typical_p": 1.0, "use_bfloat16": false, "vocab_size": 99 }, - "text_config_dict": { - "_name_or_path": "", - "add_cross_attention": false, - "architectures": null, - "attention_dropout": 0.1, - "bad_words_ids": null, - "bos_token_id": 0, - "chunk_size_feed_forward": 0, - "decoder_start_token_id": null, - "diversity_penalty": 0.0, - "do_sample": false, - "dropout": 0.1, - "early_stopping": false, - "encoder_no_repeat_ngram_size": 0, - "eos_token_id": 2, - "finetuning_task": null, - "forced_bos_token_id": null, - "forced_eos_token_id": null, - "gradient_checkpointing": false, - "hidden_act": "quick_gelu", - "hidden_size": 32, - "id2label": { - "0": "LABEL_0", - "1": "LABEL_1" - }, - "initializer_factor": 1.0, - "initializer_range": 0.02, - "intermediate_size": 37, - "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": 512, - "min_length": 0, - "model_type": "clip_text_model", - "no_repeat_ngram_size": 0, - "num_attention_heads": 4, - "num_beam_groups": 1, - "num_beams": 1, - "num_hidden_layers": 5, - "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, - "task_specific_params": null, - "temperature": 1.0, - "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.11.0.dev0", - "use_bfloat16": false, - "vocab_size": 99 - }, "torch_dtype": "float32", "transformers_version": null, "vision_config": { @@ -162,6 +94,7 @@ "architectures": null, "attention_dropout": 0.1, "bad_words_ids": null, + "begin_suppress_tokens": null, "bos_token_id": null, "chunk_size_feed_forward": 0, "cross_attention_hidden_size": null, @@ -212,12 +145,14 @@ "patch_size": 2, "prefix": null, "problem_type": null, + "projection_dim": 512, "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, @@ -228,81 +163,9 @@ "top_p": 1.0, "torch_dtype": null, "torchscript": false, - "transformers_version": "4.22.0.dev0", + "transformers_version": "4.28.0.dev0", "typical_p": 1.0, "use_bfloat16": false }, - "vision_config_dict": { - "_name_or_path": "", - "add_cross_attention": false, - "architectures": null, - "attention_dropout": 0.1, - "bad_words_ids": null, - "bos_token_id": null, - "chunk_size_feed_forward": 0, - "decoder_start_token_id": null, - "diversity_penalty": 0.0, - "do_sample": false, - "dropout": 0.1, - "early_stopping": false, - "encoder_no_repeat_ngram_size": 0, - "eos_token_id": null, - "finetuning_task": null, - "forced_bos_token_id": null, - "forced_eos_token_id": null, - "gradient_checkpointing": false, - "hidden_act": "quick_gelu", - "hidden_size": 32, - "id2label": { - "0": "LABEL_0", - "1": "LABEL_1" - }, - "image_size": 30, - "initializer_factor": 1.0, - "initializer_range": 0.02, - "intermediate_size": 37, - "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": 4, - "num_beam_groups": 1, - "num_beams": 1, - "num_channels": 3, - "num_hidden_layers": 5, - "num_return_sequences": 1, - "output_attentions": false, - "output_hidden_states": false, - "output_scores": false, - "pad_token_id": null, - "patch_size": 2, - "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, - "task_specific_params": null, - "temperature": 1.0, - "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.11.0.dev0", - "use_bfloat16": false - }, "vocab_size": 1000 } diff --git a/safety_checker/model.fp16.safetensors b/safety_checker/model.fp16.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..58e37ab35b11700e17af89458930713293be9cbc --- /dev/null +++ b/safety_checker/model.fp16.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:79596c9b7a926a48601263c914128f8ca61faad5f5c7ae75fd024c5601f044f6 +size 193428 diff --git a/safety_checker/model.safetensors b/safety_checker/model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..58e37ab35b11700e17af89458930713293be9cbc --- /dev/null +++ b/safety_checker/model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:79596c9b7a926a48601263c914128f8ca61faad5f5c7ae75fd024c5601f044f6 +size 193428 diff --git a/safety_checker/pytorch_model.fp16.bin b/safety_checker/pytorch_model.fp16.bin new file mode 100644 index 0000000000000000000000000000000000000000..90674dd533bc1e25088425857f02f18bc7791efc --- /dev/null +++ b/safety_checker/pytorch_model.fp16.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:bf963bcd70d3b007da6c367af97e3c5fa388880191e17393f1ef40f9acf90eca +size 213539 diff --git a/scheduler/scheduler_config.json b/scheduler/scheduler_config.json index 9e62adbcc1557aa252534776a786f498b6ec967e..2ea5f33fbaa86239a796a17af3be72a4dff0e29b 100644 --- a/scheduler/scheduler_config.json +++ b/scheduler/scheduler_config.json @@ -1,10 +1,17 @@ { - "_class_name": "FlaxDDIMScheduler", - "_diffusers_version": "0.4.0.dev0", + "_class_name": "DDIMScheduler", + "_diffusers_version": "0.15.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", + "sample_max_value": 1.0, + "set_alpha_to_one": true, "steps_offset": 1, - "clip_sample": false + "thresholding": false, + "trained_betas": null } diff --git a/text_encoder/model-00001-of-00023.safetensors b/text_encoder/model-00001-of-00023.safetensors index 1cd2abb9d5a71cd22474c045007fefd25cdcc911..1d6e8debe8ebdd582c7364ceb607630a062ba226 100644 Binary files a/text_encoder/model-00001-of-00023.safetensors and b/text_encoder/model-00001-of-00023.safetensors differ diff --git a/text_encoder/model-00002-of-00023.safetensors b/text_encoder/model-00002-of-00023.safetensors index 7b4043cb7625be5a82739a8c2bd6524dea555494..06fed4c99e648e4fe87d9dccb77cd1694cc575e0 100644 Binary files a/text_encoder/model-00002-of-00023.safetensors and b/text_encoder/model-00002-of-00023.safetensors differ diff --git a/text_encoder/model-00003-of-00023.safetensors b/text_encoder/model-00003-of-00023.safetensors index 8a8ec5925f7ec90df7510b787f51efcefd6180a1..d7da1d6a347213410234b1e1d3ffe632b9f5c3b2 100644 Binary files a/text_encoder/model-00003-of-00023.safetensors and b/text_encoder/model-00003-of-00023.safetensors differ diff --git a/text_encoder/model-00004-of-00023.safetensors b/text_encoder/model-00004-of-00023.safetensors index 032bd3a3689e1f5c0912efa46661f6b3874f7bf0..4463a13dcd5d3aa6d28f083d26b62d7b7d901b05 100644 Binary files a/text_encoder/model-00004-of-00023.safetensors and b/text_encoder/model-00004-of-00023.safetensors differ diff --git a/text_encoder/model-00005-of-00023.safetensors b/text_encoder/model-00005-of-00023.safetensors index e6dbdca2d244db128be7c8ad31cce6c5d02ad194..fa2a5c3b779ea3a12d1ca76c2741d5fe13384a4d 100644 Binary files a/text_encoder/model-00005-of-00023.safetensors and b/text_encoder/model-00005-of-00023.safetensors differ diff --git a/text_encoder/model-00006-of-00023.safetensors b/text_encoder/model-00006-of-00023.safetensors index 126d653276a798d3b5d8dde2e3ab9108a0b306e2..e2a03e694d0619cecd10afe67d4c4bc08b98bd97 100644 Binary files a/text_encoder/model-00006-of-00023.safetensors and b/text_encoder/model-00006-of-00023.safetensors differ diff --git a/text_encoder/model-00007-of-00023.safetensors b/text_encoder/model-00007-of-00023.safetensors index 1700b6c78d62490bac1ccf90780eddd97da87767..271d48e33f6801483c078bd0d0ee105dad5d69aa 100644 Binary files a/text_encoder/model-00007-of-00023.safetensors and b/text_encoder/model-00007-of-00023.safetensors differ diff --git a/text_encoder/model-00008-of-00023.safetensors b/text_encoder/model-00008-of-00023.safetensors index c5a60fcea303d7f66f8309b144d8f7c0d429268e..bf346ad77d8ef9e60bee354da9d00e6be011b6c0 100644 Binary files a/text_encoder/model-00008-of-00023.safetensors and b/text_encoder/model-00008-of-00023.safetensors differ diff --git a/text_encoder/model-00009-of-00023.safetensors b/text_encoder/model-00009-of-00023.safetensors index b2c46d24426ec30cee1ce0165137a820657dc18c..c71e01b8f767e4c203e283600f011005cd5e6fd3 100644 Binary files a/text_encoder/model-00009-of-00023.safetensors and b/text_encoder/model-00009-of-00023.safetensors differ diff --git a/text_encoder/model-00010-of-00023.safetensors b/text_encoder/model-00010-of-00023.safetensors index 37fe7421942089ac3bf77ebb88980c68f69fd83c..4dd22ad2ab6314a63ca69728237a206d3e6b7267 100644 Binary files a/text_encoder/model-00010-of-00023.safetensors and b/text_encoder/model-00010-of-00023.safetensors differ diff --git a/text_encoder/model-00011-of-00023.safetensors b/text_encoder/model-00011-of-00023.safetensors index 2d23d12e86ca7ce68059710bd2a0861a96614355..8ef7631e0ac21a24f2ccba86955c62d782b9622a 100644 Binary files a/text_encoder/model-00011-of-00023.safetensors and b/text_encoder/model-00011-of-00023.safetensors differ diff --git a/text_encoder/model-00012-of-00023.safetensors b/text_encoder/model-00012-of-00023.safetensors index 3cd3c9324fab612e91985a5fd8a919e81b0a6099..2745fe91b896439cf461b82b7c6773447b78f4c5 100644 Binary files a/text_encoder/model-00012-of-00023.safetensors and b/text_encoder/model-00012-of-00023.safetensors differ diff --git a/text_encoder/model-00013-of-00023.safetensors b/text_encoder/model-00013-of-00023.safetensors index 8f8fd10a1ca43ae92843f48e976c6e61a0f932b9..0e9a27d9d0132ae98ea6a1b9ae24c48b018bb501 100644 Binary files a/text_encoder/model-00013-of-00023.safetensors and b/text_encoder/model-00013-of-00023.safetensors differ diff --git a/text_encoder/model-00014-of-00023.safetensors b/text_encoder/model-00014-of-00023.safetensors index 63522321a30679dccde4d0f5b608fa90e47cc162..29decba60e75dc31dfdda3ec02d141edfeb2dcaf 100644 Binary files a/text_encoder/model-00014-of-00023.safetensors and b/text_encoder/model-00014-of-00023.safetensors differ diff --git a/text_encoder/model-00015-of-00023.safetensors b/text_encoder/model-00015-of-00023.safetensors index 8edaecc4609d838eb55c15e314c0c39fa5497c01..7aaadb199739eb5682e5d35e92c9fe92da336813 100644 Binary files a/text_encoder/model-00015-of-00023.safetensors and b/text_encoder/model-00015-of-00023.safetensors differ diff --git a/text_encoder/model-00016-of-00023.safetensors b/text_encoder/model-00016-of-00023.safetensors index f3e4cad3a5bb260b0f904bb69eab24af36be465e..b3b4c52a90a3889306cd17c25fc29246289706ea 100644 Binary files a/text_encoder/model-00016-of-00023.safetensors and b/text_encoder/model-00016-of-00023.safetensors differ diff --git a/text_encoder/model-00017-of-00023.safetensors b/text_encoder/model-00017-of-00023.safetensors index dfb8e487e10723eedc5fe50fa257a6db6d341d11..8bfed0887b89baeb5ea51b6853aa9e27b7a5aea8 100644 Binary files a/text_encoder/model-00017-of-00023.safetensors and b/text_encoder/model-00017-of-00023.safetensors differ diff --git a/text_encoder/model-00018-of-00023.safetensors b/text_encoder/model-00018-of-00023.safetensors index f0cbe6a07fd178b43b7ff821806508fc9ff99239..21a069f4766089e26e030d215d7b272500a11a0f 100644 Binary files a/text_encoder/model-00018-of-00023.safetensors and b/text_encoder/model-00018-of-00023.safetensors differ diff --git a/text_encoder/model-00019-of-00023.safetensors b/text_encoder/model-00019-of-00023.safetensors index d595acde421f752500d62707a7c40869f075f353..6894910d8c3f94eac9b86dfa95f0815d7dcec34d 100644 Binary files a/text_encoder/model-00019-of-00023.safetensors and b/text_encoder/model-00019-of-00023.safetensors differ diff --git a/text_encoder/model-00020-of-00023.safetensors b/text_encoder/model-00020-of-00023.safetensors index d447712d81163b40fc127c74ec74d98f694164ed..8967796bbc514d3069f61cc0e01d65f0dfcafd50 100644 Binary files a/text_encoder/model-00020-of-00023.safetensors and b/text_encoder/model-00020-of-00023.safetensors differ diff --git a/text_encoder/model-00021-of-00023.safetensors b/text_encoder/model-00021-of-00023.safetensors index 43fcfa215f84756e0b5712763d45f13ea26c4f59..91ade6abd4939b493e53fe04074f7409c668c2f5 100644 Binary files a/text_encoder/model-00021-of-00023.safetensors and b/text_encoder/model-00021-of-00023.safetensors differ diff --git a/text_encoder/model-00022-of-00023.safetensors b/text_encoder/model-00022-of-00023.safetensors index 672084bda07f46f02c35561ef58680f8964b0daa..c97ebce9e571943d005fdd715e6f45c678742fe8 100644 Binary files a/text_encoder/model-00022-of-00023.safetensors and b/text_encoder/model-00022-of-00023.safetensors differ diff --git a/text_encoder/model-00023-of-00023.safetensors b/text_encoder/model-00023-of-00023.safetensors index 8aff4ec4a33600d4465ace127ca8d94d2efded8c..937e85a89df2d42f28f9b8d8305e0d01e9d8c467 100644 Binary files a/text_encoder/model-00023-of-00023.safetensors and b/text_encoder/model-00023-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00001-of-00023.safetensors b/text_encoder/model.fp16-00001-of-00023.safetensors index 1cd2abb9d5a71cd22474c045007fefd25cdcc911..1d6e8debe8ebdd582c7364ceb607630a062ba226 100644 Binary files a/text_encoder/model.fp16-00001-of-00023.safetensors and b/text_encoder/model.fp16-00001-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00002-of-00023.safetensors b/text_encoder/model.fp16-00002-of-00023.safetensors index 7b4043cb7625be5a82739a8c2bd6524dea555494..06fed4c99e648e4fe87d9dccb77cd1694cc575e0 100644 Binary files a/text_encoder/model.fp16-00002-of-00023.safetensors and b/text_encoder/model.fp16-00002-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00003-of-00023.safetensors b/text_encoder/model.fp16-00003-of-00023.safetensors index 8a8ec5925f7ec90df7510b787f51efcefd6180a1..d7da1d6a347213410234b1e1d3ffe632b9f5c3b2 100644 Binary files a/text_encoder/model.fp16-00003-of-00023.safetensors and b/text_encoder/model.fp16-00003-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00004-of-00023.safetensors b/text_encoder/model.fp16-00004-of-00023.safetensors index 032bd3a3689e1f5c0912efa46661f6b3874f7bf0..4463a13dcd5d3aa6d28f083d26b62d7b7d901b05 100644 Binary files a/text_encoder/model.fp16-00004-of-00023.safetensors and b/text_encoder/model.fp16-00004-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00005-of-00023.safetensors b/text_encoder/model.fp16-00005-of-00023.safetensors index e6dbdca2d244db128be7c8ad31cce6c5d02ad194..fa2a5c3b779ea3a12d1ca76c2741d5fe13384a4d 100644 Binary files a/text_encoder/model.fp16-00005-of-00023.safetensors and b/text_encoder/model.fp16-00005-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00006-of-00023.safetensors b/text_encoder/model.fp16-00006-of-00023.safetensors index 126d653276a798d3b5d8dde2e3ab9108a0b306e2..e2a03e694d0619cecd10afe67d4c4bc08b98bd97 100644 Binary files a/text_encoder/model.fp16-00006-of-00023.safetensors and b/text_encoder/model.fp16-00006-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00007-of-00023.safetensors b/text_encoder/model.fp16-00007-of-00023.safetensors index 1700b6c78d62490bac1ccf90780eddd97da87767..271d48e33f6801483c078bd0d0ee105dad5d69aa 100644 Binary files a/text_encoder/model.fp16-00007-of-00023.safetensors and b/text_encoder/model.fp16-00007-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00008-of-00023.safetensors b/text_encoder/model.fp16-00008-of-00023.safetensors index c5a60fcea303d7f66f8309b144d8f7c0d429268e..bf346ad77d8ef9e60bee354da9d00e6be011b6c0 100644 Binary files a/text_encoder/model.fp16-00008-of-00023.safetensors and b/text_encoder/model.fp16-00008-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00009-of-00023.safetensors b/text_encoder/model.fp16-00009-of-00023.safetensors index b2c46d24426ec30cee1ce0165137a820657dc18c..c71e01b8f767e4c203e283600f011005cd5e6fd3 100644 Binary files a/text_encoder/model.fp16-00009-of-00023.safetensors and b/text_encoder/model.fp16-00009-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00010-of-00023.safetensors b/text_encoder/model.fp16-00010-of-00023.safetensors index 37fe7421942089ac3bf77ebb88980c68f69fd83c..4dd22ad2ab6314a63ca69728237a206d3e6b7267 100644 Binary files a/text_encoder/model.fp16-00010-of-00023.safetensors and b/text_encoder/model.fp16-00010-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00011-of-00023.safetensors b/text_encoder/model.fp16-00011-of-00023.safetensors index 2d23d12e86ca7ce68059710bd2a0861a96614355..8ef7631e0ac21a24f2ccba86955c62d782b9622a 100644 Binary files a/text_encoder/model.fp16-00011-of-00023.safetensors and b/text_encoder/model.fp16-00011-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00012-of-00023.safetensors b/text_encoder/model.fp16-00012-of-00023.safetensors index 3cd3c9324fab612e91985a5fd8a919e81b0a6099..2745fe91b896439cf461b82b7c6773447b78f4c5 100644 Binary files a/text_encoder/model.fp16-00012-of-00023.safetensors and b/text_encoder/model.fp16-00012-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00013-of-00023.safetensors b/text_encoder/model.fp16-00013-of-00023.safetensors index 8f8fd10a1ca43ae92843f48e976c6e61a0f932b9..0e9a27d9d0132ae98ea6a1b9ae24c48b018bb501 100644 Binary files a/text_encoder/model.fp16-00013-of-00023.safetensors and b/text_encoder/model.fp16-00013-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00014-of-00023.safetensors b/text_encoder/model.fp16-00014-of-00023.safetensors index 63522321a30679dccde4d0f5b608fa90e47cc162..29decba60e75dc31dfdda3ec02d141edfeb2dcaf 100644 Binary files a/text_encoder/model.fp16-00014-of-00023.safetensors and b/text_encoder/model.fp16-00014-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00015-of-00023.safetensors b/text_encoder/model.fp16-00015-of-00023.safetensors index 8edaecc4609d838eb55c15e314c0c39fa5497c01..7aaadb199739eb5682e5d35e92c9fe92da336813 100644 Binary files a/text_encoder/model.fp16-00015-of-00023.safetensors and b/text_encoder/model.fp16-00015-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00016-of-00023.safetensors b/text_encoder/model.fp16-00016-of-00023.safetensors index f3e4cad3a5bb260b0f904bb69eab24af36be465e..b3b4c52a90a3889306cd17c25fc29246289706ea 100644 Binary files a/text_encoder/model.fp16-00016-of-00023.safetensors and b/text_encoder/model.fp16-00016-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00017-of-00023.safetensors b/text_encoder/model.fp16-00017-of-00023.safetensors index dfb8e487e10723eedc5fe50fa257a6db6d341d11..8bfed0887b89baeb5ea51b6853aa9e27b7a5aea8 100644 Binary files a/text_encoder/model.fp16-00017-of-00023.safetensors and b/text_encoder/model.fp16-00017-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00018-of-00023.safetensors b/text_encoder/model.fp16-00018-of-00023.safetensors index f0cbe6a07fd178b43b7ff821806508fc9ff99239..21a069f4766089e26e030d215d7b272500a11a0f 100644 Binary files a/text_encoder/model.fp16-00018-of-00023.safetensors and b/text_encoder/model.fp16-00018-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00019-of-00023.safetensors b/text_encoder/model.fp16-00019-of-00023.safetensors index d595acde421f752500d62707a7c40869f075f353..6894910d8c3f94eac9b86dfa95f0815d7dcec34d 100644 Binary files a/text_encoder/model.fp16-00019-of-00023.safetensors and b/text_encoder/model.fp16-00019-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00020-of-00023.safetensors b/text_encoder/model.fp16-00020-of-00023.safetensors index d447712d81163b40fc127c74ec74d98f694164ed..8967796bbc514d3069f61cc0e01d65f0dfcafd50 100644 Binary files a/text_encoder/model.fp16-00020-of-00023.safetensors and b/text_encoder/model.fp16-00020-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00021-of-00023.safetensors b/text_encoder/model.fp16-00021-of-00023.safetensors index 43fcfa215f84756e0b5712763d45f13ea26c4f59..91ade6abd4939b493e53fe04074f7409c668c2f5 100644 Binary files a/text_encoder/model.fp16-00021-of-00023.safetensors and b/text_encoder/model.fp16-00021-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00022-of-00023.safetensors b/text_encoder/model.fp16-00022-of-00023.safetensors index 672084bda07f46f02c35561ef58680f8964b0daa..c97ebce9e571943d005fdd715e6f45c678742fe8 100644 Binary files a/text_encoder/model.fp16-00022-of-00023.safetensors and b/text_encoder/model.fp16-00022-of-00023.safetensors differ diff --git a/text_encoder/model.fp16-00023-of-00023.safetensors b/text_encoder/model.fp16-00023-of-00023.safetensors index 8aff4ec4a33600d4465ace127ca8d94d2efded8c..937e85a89df2d42f28f9b8d8305e0d01e9d8c467 100644 Binary files a/text_encoder/model.fp16-00023-of-00023.safetensors and b/text_encoder/model.fp16-00023-of-00023.safetensors differ diff --git a/tokenizer/special_tokens_map.json b/tokenizer/special_tokens_map.json index 9bfb42aa97dcd61e89f279ccaee988bccb4fabae..2c2130b544c0c5a72d5d00da071ba130a9800fb2 100644 --- a/tokenizer/special_tokens_map.json +++ b/tokenizer/special_tokens_map.json @@ -1 +1,24 @@ -{"bos_token": {"content": "<|startoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "eos_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "unk_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true}, "pad_token": "<|endoftext|>"} \ No newline at end of file +{ + "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 + } +} diff --git a/tokenizer/tokenizer_config.json b/tokenizer/tokenizer_config.json index 7532cd4b194060007a910f5796d3a45ed8015fac..d82e92f0f3618ce5503abe2d27ea2cc10193f813 100644 --- a/tokenizer/tokenizer_config.json +++ b/tokenizer/tokenizer_config.json @@ -1 +1,34 @@ -{"errors": "replace", "unk_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "bos_token": {"content": "<|startoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "eos_token": {"content": "<|endoftext|>", "single_word": false, "lstrip": false, "rstrip": false, "normalized": true, "__type": "AddedToken"}, "pad_token": "<|endoftext|>", "add_prefix_space": false, "do_lower_case": true, "model_max_length": 77, "name_or_path": "/home/patrick/.cache/huggingface/diffusers/models--hf-internal-testing--tiny-stable-diffusion-pipe/snapshots/695e29de8f0f8595542b8a204a570f44a4e9bfbb/tokenizer", "special_tokens_map_file": "/home/lysandre/.cache/huggingface/transformers/18a566598f286c9139f88160c99f84eec492a26bd22738fa9cb44d5b7e0a5c76.cce1206abbad28826f000510f22f354e53e66a97f7c23745a7dfe27609cc07f5", "tokenizer_class": "CLIPTokenizer"} \ No newline at end of file +{ + "add_prefix_space": false, + "bos_token": { + "__type": "AddedToken", + "content": "<|startoftext|>", + "lstrip": false, + "normalized": true, + "rstrip": false, + "single_word": false + }, + "clean_up_tokenization_spaces": true, + "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, + "pad_token": "<|endoftext|>", + "special_tokens_map_file": "/home/lysandre/.cache/huggingface/transformers/18a566598f286c9139f88160c99f84eec492a26bd22738fa9cb44d5b7e0a5c76.cce1206abbad28826f000510f22f354e53e66a97f7c23745a7dfe27609cc07f5", + "tokenizer_class": "CLIPTokenizer", + "unk_token": { + "__type": "AddedToken", + "content": "<|endoftext|>", + "lstrip": false, + "normalized": true, + "rstrip": false, + "single_word": false + } +} diff --git a/tokenizer/vocab.json b/tokenizer/vocab.json index 67c203f550a5c376198d25e6d2b22f9e7c008b9f..609c08ebb708079189a79c6964c3495e36b26f41 100644 --- a/tokenizer/vocab.json +++ b/tokenizer/vocab.json @@ -1 +1,1002 @@ -{"!": 2, "!": 345, "\"": 3, "\"": 344, "#": 4, "#": 325, "$": 5, "$": 348, "%": 6, "%": 351, "&": 7, "&": 352, "'": 8, "'": 296, "(": 9, "(": 318, ")": 10, ")": 330, "*": 11, "*": 327, "+": 12, "+": 341, ",": 13, ",": 279, ",@": 754, "-": 14, "-": 276, "-@": 439, ".": 15, ".": 253, ".@": 695, "/": 16, "/": 350, "0": 17, "00": 647, "0": 216, "1": 18, "1": 222, "2": 19, "2": 231, "3": 20, "3": 243, "4": 21, "4": 233, "5": 22, "5": 240, "6": 23, "6": 226, "7": 24, "7": 215, "8": 25, "8": 236, "9": 26, "9": 242, ":": 27, ":": 353, ";": 28, ";": 317, "<": 29, "<": 340, "<|endoftext|>": 1, "<|startoftext|>": 0, "=": 30, "=": 342, ">": 31, ">": 300, "?": 32, "?": 346, "@": 33, "@": 320, "A": 34, "A": 227, "B": 35, "B": 258, "C": 36, "C": 239, "D": 37, "D": 255, "E": 38, "E": 246, "F": 39, "F": 213, "G": 40, "G": 283, "H": 41, "H": 219, "I": 42, "I": 237, "J": 43, "J": 251, "K": 44, "K": 254, "L": 45, "L": 218, "M": 46, "M": 234, "N": 47, "N": 238, "O": 48, "O": 265, "P": 49, "P": 245, "Q": 50, "Q": 309, "R": 51, "R": 264, "S": 52, "S": 230, "T": 53, "T": 235, "U": 54, "U": 268, "V": 55, "V": 248, "W": 56, "W": 274, "X": 57, "X": 263, "Y": 58, "Y": 310, "Z": 59, "Z": 207, "[": 60, "[": 270, "\\": 61, "\\": 338, "]": 62, "]": 289, "^": 63, "^": 331, "_": 64, "_": 334, "`": 65, "`": 347, "a": 66, "a": 197, "ab": 555, "able": 820, "ac": 420, "ace": 806, "ach": 791, "ach": 885, "ack": 670, "act": 929, "ad": 508, "ad": 860, "ade": 771, "ag": 511, "age": 710, "ain": 568, "ain": 675, "ak": 577, "ake": 882, "al": 397, "al": 405, "all": 664, "all": 658, "ally": 588, "als": 796, "am": 426, "am": 817, "ame": 552, "ames": 976, "amp": 800, "an": 384, "an": 425, "ance": 751, "and": 609, "and": 780, "ang": 816, "ans": 844, "ant": 837, "ant": 753, "any": 766, "ap": 586, "ar": 376, "ar": 579, "ard": 649, "ards": 982, "ary": 611, "as": 416, "as": 404, "ase": 849, "ased": 814, "ason": 865, "ass": 792, "ast": 661, "at": 372, "at": 434, "ate": 541, "ated": 543, "ater": 709, "ates": 825, "ath": 730, "ating": 922, "ation": 497, "ational": 933, "ations": 744, "att": 903, "aus": 858, "av": 681, "ay": 684, "ay": 523, "b": 67, "b": 212, "ber": 593, "c": 68, "c": 224, "cc": 960, "ce": 496, "ces": 830, "ch": 520, "ch": 603, "ct": 834, "d": 69, "d": 196, "ded": 665, "der": 690, "ding": 633, "ds": 530, "duc": 671, "e": 70, "e": 195, "ea": 471, "ear": 596, "ear": 669, "ears": 906, "eb": 852, "ec": 418, "ect": 838, "ect": 964, "ed": 563, "ed": 362, "ee": 941, "een": 779, "ef": 840, "eg": 731, "el": 407, "el": 610, "eld": 973, "ell": 759, "ell": 756, "ely": 719, "em": 455, "ember": 777, "ement": 959, "emp": 975, "en": 375, "en": 427, "ence": 685, "ens": 880, "ent": 478, "ent": 468, "ents": 674, "ep": 545, "er": 364, "er": 374, "eral": 793, "ere": 481, "ered": 748, "eric": 921, "erm": 861, "ern": 887, "ern": 977, "ers": 598, "ers": 486, "ert": 986, "ery": 805, "es": 402, "es": 388, "ese": 794, "ess": 678, "ess": 693, "est": 606, "est": 584, "et": 460, "et": 594, "etw": 824, "etween": 886, "ev": 493, "evel": 980, "ever": 855, "ew": 687, "ew": 612, "ex": 938, "ey": 713, "f": 71, "f": 209, "fer": 911, "ff": 587, "for": 728, "form": 901, "fter": 634, "g": 72, "g": 214, "ge": 592, "h": 73, "h": 203, "i": 74, "i": 205, "ia": 605, "ial": 672, "ian": 638, "ib": 726, "ic": 395, "ic": 510, "ical": 625, "ice": 782, "ich": 561, "ics": 996, "id": 463, "id": 613, "ide": 739, "ie": 974, "ied": 812, "ies": 516, "if": 524, "ig": 444, "igh": 537, "ight": 680, "ik": 775, "ike": 984, "il": 406, "il": 714, "ile": 721, "ill": 608, "ill": 789, "ily": 950, "im": 469, "im": 767, "ime": 691, "in": 358, "in": 501, "ine": 607, "ing": 557, "ing": 383, "ings": 815, "ion": 472, "ion": 408, "ional": 717, "ions": 540, "ip": 733, "ip": 818, "ir": 453, "ir": 554, "is": 393, "is": 441, "ish": 694, "ish": 654, "ished": 942, "ision": 944, "iss": 876, "ist": 550, "ist": 811, "it": 378, "it": 746, "ite": 760, "ited": 809, "ition": 797, "ity": 542, "iv": 435, "ive": 549, "ived": 979, "iz": 722, "j": 75, "j": 288, "k": 76, "k": 210, "ked": 810, "king": 924, "ks": 692, "l": 77, "l": 201, "la": 467, "land": 743, "ld": 559, "le": 536, "le": 465, "les": 799, "lud": 718, "ly": 433, "m": 78, "m": 202, "ment": 701, "mp": 651, "n": 79, "n": 199, "nd": 369, "ned": 758, "ning": 843, "o": 80, "o": 198, "ob": 920, "oc": 534, "od": 575, "og": 604, "oh": 972, "oin": 831, "ol": 428, "oll": 703, "ollow": 928, "olog": 932, "om": 419, "om": 883, "ome": 663, "on": 382, "on": 390, "ond": 872, "one": 835, "ong": 850, "ong": 582, "oo": 517, "ood": 927, "ook": 897, "op": 531, "op": 971, "or": 377, "or": 424, "ore": 571, "ors": 917, "ort": 768, "ort": 752, "ory": 737, "os": 447, "ose": 881, "ost": 646, "ot": 600, "ot": 879, "ou": 392, "oug": 659, "ough": 798, "ould": 640, "oun": 553, "ound": 961, "our": 648, "our": 772, "ous": 712, "out": 683, "outh": 945, "ov": 515, "ow": 461, "ow": 666, "own": 657, "oy": 952, "p": 81, "p": 217, "per": 715, "ph": 916, "pp": 518, "q": 82, "q": 280, "qu": 546, "r": 83, "r": 204, "ra": 457, "ran": 624, "re": 367, "ree": 765, "ren": 790, "res": 572, "res": 747, "ri": 487, "rib": 804, "ric": 745, "rit": 589, "ro": 385, "rom": 498, "rop": 826, "roug": 803, "ru": 951, "ruc": 891, "ry": 908, "s": 84, "s": 206, "se": 741, "sh": 795, "so": 630, "sp": 992, "ss": 673, "st": 519, "st": 528, "t": 85, "t": 208, "te": 954, "ted": 489, "ter": 535, "ter": 505, "th": 449, "th": 488, "ther": 576, "ting": 676, "tion": 570, "tr": 619, "ts": 436, "tt": 720, "tur": 953, "ty": 821, "u": 86, "u": 229, "ub": 591, "ubl": 842, "uc": 490, "ud": 538, "ue": 652, "ug": 560, "uil": 930, "ul": 494, "um": 532, "um": 644, "un": 448, "und": 828, "up": 833, "up": 700, "ur": 413, "ure": 635, "uring": 702, "ury": 957, "us": 438, "us": 622, "ust": 846, "ut": 529, "ut": 527, "v": 87, "v": 232, "ve": 567, "vi": 866, "w": 88, "w": 250, "way": 970, "wn": 999, "x": 89, "x": 269, "y": 90, "y": 211, "yp": 993, "z": 91, "z": 228, "|": 92, "|": 304, "}": 93, "}": 336, "~": 94, "~": 343, "¡": 95, "¡": 220, "¢": 96, "¢": 306, "£": 97, "£": 323, "¤": 98, "¤": 292, "¥": 99, "¥": 339, "¦": 100, "¦": 303, "§": 101, "§": 275, "¨": 102, "¨": 282, "©": 103, "©": 259, "ª": 104, "ª": 286, "«": 105, "«": 266, "¬": 106, "¬": 319, "®": 107, "®": 329, "¯": 108, "¯": 287, "°": 109, "°": 298, "±": 110, "±": 200, "²": 111, "²": 284, "³": 112, "³": 272, "´": 113, "´": 307, "µ": 114, "µ": 261, "¶": 115, "¶": 301, "·": 116, "·": 326, "¸": 117, "¸": 257, "¹": 118, "¹": 241, "º": 119, "º": 260, "»": 120, "»": 247, "¼": 121, "¼": 305, "½": 122, "½": 294, "¾": 123, "¾": 316, "¿": 124, "¿": 271, "Â": 125, "Ã": 126, "Ä": 127, "Å": 128, "Æ": 129, "Ç": 130, "È": 131, "É": 132, "Ê": 133, "Ë": 134, "Ì": 135, "Í": 136, "Î": 137, "Ï": 138, "Ð": 139, "Ñ": 140, "Ö": 141, "×": 142, "Ø": 143, "Ù": 144, "Ü": 145, "à": 146, "á": 147, "â": 148, "ã": 149, "ä": 150, "å": 151, "æ": 152, "ç": 153, "è": 154, "é": 155, "ë": 156, "ì": 157, "ï": 158, "Ċ": 159, "Ċ": 349, "Ġ": 160, "Ġ\"": 401, "Ġ'": 431, "Ġ(": 475, "Ġ)": 474, "Ġ,": 360, "Ġ.": 365, "Ġ0": 847, "Ġ1": 411, "Ġ18": 769, "Ġ19": 492, "Ġ199": 893, "Ġ1": 778, "Ġ2": 462, "Ġ20": 522, "Ġ200": 620, "Ġ201": 734, "Ġ2": 813, "Ġ3": 735, "Ġ3": 888, "Ġ4": 870, "Ġ5": 907, "Ġ5": 990, "Ġ:": 637, "Ġ;": 615, "Ġ": 333, "Ġ=": 399, "Ġ@": 417, "Ġ@,@": 755, "Ġ@-@": 440, "Ġ@.@": 696, "ĠA": 409, "ĠA": 807, "ĠAl": 716, "ĠAm": 829, "ĠAmeric": 958, "ĠAn": 784, "ĠAr": 894, "ĠB": 432, "ĠC": 410, "ĠCh": 581, "ĠCom": 904, "ĠD": 464, "ĠE": 500, "ĠEn": 878, "ĠF": 470, "ĠG": 482, "ĠH": 445, "ĠHe": 742, "ĠI": 442, "ĠI": 827, "ĠIn": 704, "ĠIn": 574, "ĠIt": 774, "ĠJ": 491, "ĠK": 548, "ĠL": 484, "ĠM": 423, "ĠMar": 776, "ĠN": 483, "ĠO": 504, "ĠP": 450, "ĠPar": 967, "ĠR": 459, "ĠS": 403, "ĠSh": 750, "ĠSt": 590, "ĠT": 396, "ĠTh": 414, "ĠThe": 437, "ĠThis": 997, "ĠU": 585, "ĠUn": 773, "ĠV": 617, "ĠW": 479, "ĠWh": 853, "ĠY": 757, "Ġa": 356, "Ġa": 394, "Ġab": 653, "Ġabout": 899, "Ġac": 583, "Ġacc": 874, "Ġad": 656, "Ġafter": 763, "Ġag": 725, "Ġal": 476, "Ġalb": 991, "Ġall": 839, "Ġalso": 641, "Ġan": 602, "Ġan": 562, "Ġand": 381, "Ġapp": 711, "Ġar": 507, "Ġare": 601, "Ġas": 454, "Ġass": 947, "Ġat": 514, "Ġatt": 788, "Ġb": 371, "Ġbe": 499, "Ġbe": 595, "Ġbec": 706, "Ġbeen": 686, "Ġbeg": 915, "Ġbetween": 914, "Ġbo": 819, "Ġbut": 623, "Ġby": 473, "Ġc": 368, "Ġcent": 823, "Ġch": 526, "Ġchar": 822, "Ġcl": 689, "Ġcom": 509, "Ġcomm": 707, "Ġcomp": 616, "Ġcon": 477, "Ġcons": 841, "Ġcont": 655, "Ġcre": 931, "Ġd": 387, "Ġde": 627, "Ġdec": 873, "Ġdef": 965, "Ġdes": 738, "Ġdi": 892, "Ġdis": 708, "Ġduring": 864, "Ġe": 421, "Ġear": 854, "Ġel": 869, "Ġen": 662, "Ġev": 682, "Ġex": 539, "Ġexp": 867, "Ġf": 370, "Ġfe": 845, "Ġfil": 913, "Ġfin": 786, "Ġfir": 599, "Ġfirst": 626, "Ġfl": 877, "Ġfor": 614, "Ġfor": 458, "Ġform": 963, "Ġfrom": 503, "Ġg": 430, "Ġgame": 895, "Ġgen": 985, "Ġgro": 890, "Ġh": 380, "Ġha": 485, "Ġhad": 566, "Ġhas": 679, "Ġhave": 667, "Ġhe": 558, "Ġher": 660, "Ġhim": 896, "Ġhis": 512, "Ġi": 366, "Ġimp": 909, "Ġin": 429, "Ġin": 389, "Ġinc": 618, "Ġinclud": 761, "Ġind": 983, "Ġint": 628, "Ġinter": 832, "Ġinto": 785, "Ġis": 480, "Ġit": 533, "Ġits": 642, "Ġj": 723, "Ġk": 564, "Ġkn": 857, "Ġl": 398, "Ġlar": 962, "Ġlater": 936, "Ġlea": 868, "Ġm": 386, "Ġmade": 949, "Ġme": 968, "Ġmore": 802, "Ġmost": 910, "Ġmov": 956, "Ġmus": 966, "Ġn": 415, "Ġnew": 989, "Ġno": 547, "Ġnor": 978, "Ġnot": 632, "Ġnum": 926, "Ġo": 359, "Ġof": 373, "Ġoff": 875, "Ġon": 551, "Ġon": 456, "Ġone": 677, "Ġonly": 871, "Ġor": 699, "Ġor": 645, "Ġother": 787, "Ġout": 925, "Ġov": 729, "Ġover": 856, "Ġp": 379, "Ġpar": 636, "Ġper": 736, "Ġpl": 698, "Ġpla": 697, "Ġplay": 808, "Ġpos": 859, "Ġpr": 889, "Ġpre": 749, "Ġpres": 912, "Ġpro": 506, "Ġproduc": 934, "Ġqu": 955, "Ġr": 521, "Ġra": 863, "Ġre": 400, "Ġrec": 597, "Ġrecor": 919, "Ġreg": 943, "Ġrel": 900, "Ġrele": 946, "Ġrem": 848, "Ġrep": 762, "Ġres": 650, "Ġro": 629, "Ġs": 361, "Ġsa": 905, "Ġsc": 732, "Ġse": 569, "Ġseason": 948, "Ġsec": 781, "Ġser": 740, "Ġsev": 884, "Ġsh": 513, "Ġshe": 862, "Ġsp": 578, "Ġspec": 940, "Ġst": 446, "Ġstar": 939, "Ġsu": 770, "Ġsub": 969, "Ġsuc": 764, "Ġsuch": 981, "Ġt": 354, "Ġth": 355, "Ġthan": 918, "Ġthat": 452, "Ġthe": 502, "Ġthe": 357, "Ġtheir": 621, "Ġthem": 998, "Ġthey": 727, "Ġthis": 705, "Ġthree": 902, "Ġthroug": 923, "Ġtime": 783, "Ġto": 391, "Ġtra": 836, "Ġtw": 639, "Ġtwo": 688, "Ġun": 544, "Ġup": 898, "Ġus": 668, "Ġused": 988, "Ġv": 495, "Ġw": 363, "Ġwas": 422, "Ġwere": 525, "Ġwh": 443, "Ġwhen": 851, "Ġwhere": 995, "Ġwhich": 573, "Ġwhile": 935, "Ġwho": 724, "Ġwit": 451, "Ġwith": 994, "Ġwith": 466, "Ġwor": 643, "Ġwould": 801, "Ġwrit": 937, "Ġy": 580, "Ġyear": 987, "Ġâ": 556, "ĠâĢ": 565, "ĠâĢĵ": 631, "ĠĊ": 412, "Ģ": 161, "Ģ": 223, "ģ": 162, "ģ": 273, "Ĥ": 163, "Ĥ": 262, "ĥ": 164, "ĥ": 337, "Ħ": 165, "Ħ": 278, "ħ": 166, "ħ": 281, "Ĩ": 167, "Ĩ": 308, "ĩ": 168, "ĩ": 225, "Ī": 169, "Ī": 221, "ī": 170, "ī": 244, "Ĭ": 171, "Ĭ": 315, "ĭ": 172, "ĭ": 321, "Į": 173, "Į": 324, "į": 174, "į": 302, "İ": 175, "İ": 249, "ı": 176, "ı": 332, "IJ": 177, "IJ": 295, "ij": 178, "ij": 313, "Ĵ": 179, "Ĵ": 328, "ĵ": 180, "ĵ": 312, "Ķ": 181, "Ķ": 256, "ķ": 182, "ķ": 314, "ĸ": 183, "ĸ": 277, "Ĺ": 184, "Ĺ": 322, "ĺ": 185, "ĺ": 285, "Ļ": 186, "Ļ": 267, "ļ": 187, "ļ": 290, "Ľ": 188, "Ľ": 311, "ľ": 189, "ľ": 299, "Ŀ": 190, "Ŀ": 291, "ŀ": 191, "ŀ": 293, "Ł": 192, "Ł": 335, "ł": 193, "ł": 252, "Ń": 194, "Ń": 297} \ No newline at end of file +{ + "!": 2, + "!": 345, + "\"": 3, + "\"": 344, + "#": 4, + "#": 325, + "$": 5, + "$": 348, + "%": 6, + "%": 351, + "&": 7, + "&": 352, + "'": 8, + "'": 296, + "(": 9, + "(": 318, + ")": 10, + ")": 330, + "*": 11, + "*": 327, + "+": 12, + "+": 341, + ",": 13, + ",": 279, + ",@": 754, + "-": 14, + "-": 276, + "-@": 439, + ".": 15, + ".": 253, + ".@": 695, + "/": 16, + "/": 350, + "0": 17, + "00": 647, + "0": 216, + "1": 18, + "1": 222, + "2": 19, + "2": 231, + "3": 20, + "3": 243, + "4": 21, + "4": 233, + "5": 22, + "5": 240, + "6": 23, + "6": 226, + "7": 24, + "7": 215, + "8": 25, + "8": 236, + "9": 26, + "9": 242, + ":": 27, + ":": 353, + ";": 28, + ";": 317, + "<": 29, + "<": 340, + "<|endoftext|>": 1, + "<|startoftext|>": 0, + "=": 30, + "=": 342, + ">": 31, + ">": 300, + "?": 32, + "?": 346, + "@": 33, + "@": 320, + "A": 34, + "A": 227, + "B": 35, + "B": 258, + "C": 36, + "C": 239, + "D": 37, + "D": 255, + "E": 38, + "E": 246, + "F": 39, + "F": 213, + "G": 40, + "G": 283, + "H": 41, + "H": 219, + "I": 42, + "I": 237, + "J": 43, + "J": 251, + "K": 44, + "K": 254, + "L": 45, + "L": 218, + "M": 46, + "M": 234, + "N": 47, + "N": 238, + "O": 48, + "O": 265, + "P": 49, + "P": 245, + "Q": 50, + "Q": 309, + "R": 51, + "R": 264, + "S": 52, + "S": 230, + "T": 53, + "T": 235, + "U": 54, + "U": 268, + "V": 55, + "V": 248, + "W": 56, + "W": 274, + "X": 57, + "X": 263, + "Y": 58, + "Y": 310, + "Z": 59, + "Z": 207, + "[": 60, + "[": 270, + "\\": 61, + "\\": 338, + "]": 62, + "]": 289, + "^": 63, + "^": 331, + "_": 64, + "_": 334, + "`": 65, + "`": 347, + "a": 66, + "a": 197, + "ab": 555, + "able": 820, + "ac": 420, + "ace": 806, + "ach": 791, + "ach": 885, + "ack": 670, + "act": 929, + "ad": 508, + "ad": 860, + "ade": 771, + "ag": 511, + "age": 710, + "ain": 568, + "ain": 675, + "ak": 577, + "ake": 882, + "al": 397, + "al": 405, + "all": 664, + "all": 658, + "ally": 588, + "als": 796, + "am": 426, + "am": 817, + "ame": 552, + "ames": 976, + "amp": 800, + "an": 384, + "an": 425, + "ance": 751, + "and": 609, + "and": 780, + "ang": 816, + "ans": 844, + "ant": 837, + "ant": 753, + "any": 766, + "ap": 586, + "ar": 376, + "ar": 579, + "ard": 649, + "ards": 982, + "ary": 611, + "as": 416, + "as": 404, + "ase": 849, + "ased": 814, + "ason": 865, + "ass": 792, + "ast": 661, + "at": 372, + "at": 434, + "ate": 541, + "ated": 543, + "ater": 709, + "ates": 825, + "ath": 730, + "ating": 922, + "ation": 497, + "ational": 933, + "ations": 744, + "att": 903, + "aus": 858, + "av": 681, + "ay": 684, + "ay": 523, + "b": 67, + "b": 212, + "ber": 593, + "c": 68, + "c": 224, + "cc": 960, + "ce": 496, + "ces": 830, + "ch": 520, + "ch": 603, + "ct": 834, + "d": 69, + "d": 196, + "ded": 665, + "der": 690, + "ding": 633, + "ds": 530, + "duc": 671, + "e": 70, + "e": 195, + "ea": 471, + "ear": 596, + "ear": 669, + "ears": 906, + "eb": 852, + "ec": 418, + "ect": 838, + "ect": 964, + "ed": 563, + "ed": 362, + "ee": 941, + "een": 779, + "ef": 840, + "eg": 731, + "el": 407, + "el": 610, + "eld": 973, + "ell": 759, + "ell": 756, + "ely": 719, + "em": 455, + "ember": 777, + "ement": 959, + "emp": 975, + "en": 375, + "en": 427, + "ence": 685, + "ens": 880, + "ent": 478, + "ent": 468, + "ents": 674, + "ep": 545, + "er": 364, + "er": 374, + "eral": 793, + "ere": 481, + "ered": 748, + "eric": 921, + "erm": 861, + "ern": 887, + "ern": 977, + "ers": 598, + "ers": 486, + "ert": 986, + "ery": 805, + "es": 402, + "es": 388, + "ese": 794, + "ess": 678, + "ess": 693, + "est": 606, + "est": 584, + "et": 460, + "et": 594, + "etw": 824, + "etween": 886, + "ev": 493, + "evel": 980, + "ever": 855, + "ew": 687, + "ew": 612, + "ex": 938, + "ey": 713, + "f": 71, + "f": 209, + "fer": 911, + "ff": 587, + "for": 728, + "form": 901, + "fter": 634, + "g": 72, + "g": 214, + "ge": 592, + "h": 73, + "h": 203, + "i": 74, + "i": 205, + "ia": 605, + "ial": 672, + "ian": 638, + "ib": 726, + "ic": 395, + "ic": 510, + "ical": 625, + "ice": 782, + "ich": 561, + "ics": 996, + "id": 463, + "id": 613, + "ide": 739, + "ie": 974, + "ied": 812, + "ies": 516, + "if": 524, + "ig": 444, + "igh": 537, + "ight": 680, + "ik": 775, + "ike": 984, + "il": 406, + "il": 714, + "ile": 721, + "ill": 608, + "ill": 789, + "ily": 950, + "im": 469, + "im": 767, + "ime": 691, + "in": 358, + "in": 501, + "ine": 607, + "ing": 557, + "ing": 383, + "ings": 815, + "ion": 472, + "ion": 408, + "ional": 717, + "ions": 540, + "ip": 733, + "ip": 818, + "ir": 453, + "ir": 554, + "is": 393, + "is": 441, + "ish": 694, + "ish": 654, + "ished": 942, + "ision": 944, + "iss": 876, + "ist": 550, + "ist": 811, + "it": 378, + "it": 746, + "ite": 760, + "ited": 809, + "ition": 797, + "ity": 542, + "iv": 435, + "ive": 549, + "ived": 979, + "iz": 722, + "j": 75, + "j": 288, + "k": 76, + "k": 210, + "ked": 810, + "king": 924, + "ks": 692, + "l": 77, + "l": 201, + "la": 467, + "land": 743, + "ld": 559, + "le": 536, + "le": 465, + "les": 799, + "lud": 718, + "ly": 433, + "m": 78, + "m": 202, + "ment": 701, + "mp": 651, + "n": 79, + "n": 199, + "nd": 369, + "ned": 758, + "ning": 843, + "o": 80, + "o": 198, + "ob": 920, + "oc": 534, + "od": 575, + "og": 604, + "oh": 972, + "oin": 831, + "ol": 428, + "oll": 703, + "ollow": 928, + "olog": 932, + "om": 419, + "om": 883, + "ome": 663, + "on": 382, + "on": 390, + "ond": 872, + "one": 835, + "ong": 850, + "ong": 582, + "oo": 517, + "ood": 927, + "ook": 897, + "op": 531, + "op": 971, + "or": 377, + "or": 424, + "ore": 571, + "ors": 917, + "ort": 768, + "ort": 752, + "ory": 737, + "os": 447, + "ose": 881, + "ost": 646, + "ot": 600, + "ot": 879, + "ou": 392, + "oug": 659, + "ough": 798, + "ould": 640, + "oun": 553, + "ound": 961, + "our": 648, + "our": 772, + "ous": 712, + "out": 683, + "outh": 945, + "ov": 515, + "ow": 461, + "ow": 666, + "own": 657, + "oy": 952, + "p": 81, + "p": 217, + "per": 715, + "ph": 916, + "pp": 518, + "q": 82, + "q": 280, + "qu": 546, + "r": 83, + "r": 204, + "ra": 457, + "ran": 624, + "re": 367, + "ree": 765, + "ren": 790, + "res": 572, + "res": 747, + "ri": 487, + "rib": 804, + "ric": 745, + "rit": 589, + "ro": 385, + "rom": 498, + "rop": 826, + "roug": 803, + "ru": 951, + "ruc": 891, + "ry": 908, + "s": 84, + "s": 206, + "se": 741, + "sh": 795, + "so": 630, + "sp": 992, + "ss": 673, + "st": 519, + "st": 528, + "t": 85, + "t": 208, + "te": 954, + "ted": 489, + "ter": 535, + "ter": 505, + "th": 449, + "th": 488, + "ther": 576, + "ting": 676, + "tion": 570, + "tr": 619, + "ts": 436, + "tt": 720, + "tur": 953, + "ty": 821, + "u": 86, + "u": 229, + "ub": 591, + "ubl": 842, + "uc": 490, + "ud": 538, + "ue": 652, + "ug": 560, + "uil": 930, + "ul": 494, + "um": 532, + "um": 644, + "un": 448, + "und": 828, + "up": 833, + "up": 700, + "ur": 413, + "ure": 635, + "uring": 702, + "ury": 957, + "us": 438, + "us": 622, + "ust": 846, + "ut": 529, + "ut": 527, + "v": 87, + "v": 232, + "ve": 567, + "vi": 866, + "w": 88, + "w": 250, + "way": 970, + "wn": 999, + "x": 89, + "x": 269, + "y": 90, + "y": 211, + "yp": 993, + "z": 91, + "z": 228, + "|": 92, + "|": 304, + "}": 93, + "}": 336, + "~": 94, + "~": 343, + "¡": 95, + "¡": 220, + "¢": 96, + "¢": 306, + "£": 97, + "£": 323, + "¤": 98, + "¤": 292, + "¥": 99, + "¥": 339, + "¦": 100, + "¦": 303, + "§": 101, + "§": 275, + "¨": 102, + "¨": 282, + "©": 103, + "©": 259, + "ª": 104, + "ª": 286, + "«": 105, + "«": 266, + "¬": 106, + "¬": 319, + "®": 107, + "®": 329, + "¯": 108, + "¯": 287, + "°": 109, + "°": 298, + "±": 110, + "±": 200, + "²": 111, + "²": 284, + "³": 112, + "³": 272, + "´": 113, + "´": 307, + "µ": 114, + "µ": 261, + "¶": 115, + "¶": 301, + "·": 116, + "·": 326, + "¸": 117, + "¸": 257, + "¹": 118, + "¹": 241, + "º": 119, + "º": 260, + "»": 120, + "»": 247, + "¼": 121, + "¼": 305, + "½": 122, + "½": 294, + "¾": 123, + "¾": 316, + "¿": 124, + "¿": 271, + "Â": 125, + "Ã": 126, + "Ä": 127, + "Å": 128, + "Æ": 129, + "Ç": 130, + "È": 131, + "É": 132, + "Ê": 133, + "Ë": 134, + "Ì": 135, + "Í": 136, + "Î": 137, + "Ï": 138, + "Ð": 139, + "Ñ": 140, + "Ö": 141, + "×": 142, + "Ø": 143, + "Ù": 144, + "Ü": 145, + "à": 146, + "á": 147, + "â": 148, + "ã": 149, + "ä": 150, + "å": 151, + "æ": 152, + "ç": 153, + "è": 154, + "é": 155, + "ë": 156, + "ì": 157, + "ï": 158, + "Ċ": 159, + "Ċ": 349, + "Ġ": 160, + "Ġ\"": 401, + "Ġ'": 431, + "Ġ(": 475, + "Ġ)": 474, + "Ġ,": 360, + "Ġ.": 365, + "Ġ0": 847, + "Ġ1": 411, + "Ġ18": 769, + "Ġ19": 492, + "Ġ199": 893, + "Ġ1": 778, + "Ġ2": 462, + "Ġ20": 522, + "Ġ200": 620, + "Ġ201": 734, + "Ġ2": 813, + "Ġ3": 735, + "Ġ3": 888, + "Ġ4": 870, + "Ġ5": 907, + "Ġ5": 990, + "Ġ:": 637, + "Ġ;": 615, + "Ġ": 333, + "Ġ=": 399, + "Ġ@": 417, + "Ġ@,@": 755, + "Ġ@-@": 440, + "Ġ@.@": 696, + "ĠA": 409, + "ĠA": 807, + "ĠAl": 716, + "ĠAm": 829, + "ĠAmeric": 958, + "ĠAn": 784, + "ĠAr": 894, + "ĠB": 432, + "ĠC": 410, + "ĠCh": 581, + "ĠCom": 904, + "ĠD": 464, + "ĠE": 500, + "ĠEn": 878, + "ĠF": 470, + "ĠG": 482, + "ĠH": 445, + "ĠHe": 742, + "ĠI": 442, + "ĠI": 827, + "ĠIn": 704, + "ĠIn": 574, + "ĠIt": 774, + "ĠJ": 491, + "ĠK": 548, + "ĠL": 484, + "ĠM": 423, + "ĠMar": 776, + "ĠN": 483, + "ĠO": 504, + "ĠP": 450, + "ĠPar": 967, + "ĠR": 459, + "ĠS": 403, + "ĠSh": 750, + "ĠSt": 590, + "ĠT": 396, + "ĠTh": 414, + "ĠThe": 437, + "ĠThis": 997, + "ĠU": 585, + "ĠUn": 773, + "ĠV": 617, + "ĠW": 479, + "ĠWh": 853, + "ĠY": 757, + "Ġa": 356, + "Ġa": 394, + "Ġab": 653, + "Ġabout": 899, + "Ġac": 583, + "Ġacc": 874, + "Ġad": 656, + "Ġafter": 763, + "Ġag": 725, + "Ġal": 476, + "Ġalb": 991, + "Ġall": 839, + "Ġalso": 641, + "Ġan": 602, + "Ġan": 562, + "Ġand": 381, + "Ġapp": 711, + "Ġar": 507, + "Ġare": 601, + "Ġas": 454, + "Ġass": 947, + "Ġat": 514, + "Ġatt": 788, + "Ġb": 371, + "Ġbe": 499, + "Ġbe": 595, + "Ġbec": 706, + "Ġbeen": 686, + "Ġbeg": 915, + "Ġbetween": 914, + "Ġbo": 819, + "Ġbut": 623, + "Ġby": 473, + "Ġc": 368, + "Ġcent": 823, + "Ġch": 526, + "Ġchar": 822, + "Ġcl": 689, + "Ġcom": 509, + "Ġcomm": 707, + "Ġcomp": 616, + "Ġcon": 477, + "Ġcons": 841, + "Ġcont": 655, + "Ġcre": 931, + "Ġd": 387, + "Ġde": 627, + "Ġdec": 873, + "Ġdef": 965, + "Ġdes": 738, + "Ġdi": 892, + "Ġdis": 708, + "Ġduring": 864, + "Ġe": 421, + "Ġear": 854, + "Ġel": 869, + "Ġen": 662, + "Ġev": 682, + "Ġex": 539, + "Ġexp": 867, + "Ġf": 370, + "Ġfe": 845, + "Ġfil": 913, + "Ġfin": 786, + "Ġfir": 599, + "Ġfirst": 626, + "Ġfl": 877, + "Ġfor": 614, + "Ġfor": 458, + "Ġform": 963, + "Ġfrom": 503, + "Ġg": 430, + "Ġgame": 895, + "Ġgen": 985, + "Ġgro": 890, + "Ġh": 380, + "Ġha": 485, + "Ġhad": 566, + "Ġhas": 679, + "Ġhave": 667, + "Ġhe": 558, + "Ġher": 660, + "Ġhim": 896, + "Ġhis": 512, + "Ġi": 366, + "Ġimp": 909, + "Ġin": 429, + "Ġin": 389, + "Ġinc": 618, + "Ġinclud": 761, + "Ġind": 983, + "Ġint": 628, + "Ġinter": 832, + "Ġinto": 785, + "Ġis": 480, + "Ġit": 533, + "Ġits": 642, + "Ġj": 723, + "Ġk": 564, + "Ġkn": 857, + "Ġl": 398, + "Ġlar": 962, + "Ġlater": 936, + "Ġlea": 868, + "Ġm": 386, + "Ġmade": 949, + "Ġme": 968, + "Ġmore": 802, + "Ġmost": 910, + "Ġmov": 956, + "Ġmus": 966, + "Ġn": 415, + "Ġnew": 989, + "Ġno": 547, + "Ġnor": 978, + "Ġnot": 632, + "Ġnum": 926, + "Ġo": 359, + "Ġof": 373, + "Ġoff": 875, + "Ġon": 551, + "Ġon": 456, + "Ġone": 677, + "Ġonly": 871, + "Ġor": 699, + "Ġor": 645, + "Ġother": 787, + "Ġout": 925, + "Ġov": 729, + "Ġover": 856, + "Ġp": 379, + "Ġpar": 636, + "Ġper": 736, + "Ġpl": 698, + "Ġpla": 697, + "Ġplay": 808, + "Ġpos": 859, + "Ġpr": 889, + "Ġpre": 749, + "Ġpres": 912, + "Ġpro": 506, + "Ġproduc": 934, + "Ġqu": 955, + "Ġr": 521, + "Ġra": 863, + "Ġre": 400, + "Ġrec": 597, + "Ġrecor": 919, + "Ġreg": 943, + "Ġrel": 900, + "Ġrele": 946, + "Ġrem": 848, + "Ġrep": 762, + "Ġres": 650, + "Ġro": 629, + "Ġs": 361, + "Ġsa": 905, + "Ġsc": 732, + "Ġse": 569, + "Ġseason": 948, + "Ġsec": 781, + "Ġser": 740, + "Ġsev": 884, + "Ġsh": 513, + "Ġshe": 862, + "Ġsp": 578, + "Ġspec": 940, + "Ġst": 446, + "Ġstar": 939, + "Ġsu": 770, + "Ġsub": 969, + "Ġsuc": 764, + "Ġsuch": 981, + "Ġt": 354, + "Ġth": 355, + "Ġthan": 918, + "Ġthat": 452, + "Ġthe": 502, + "Ġthe": 357, + "Ġtheir": 621, + "Ġthem": 998, + "Ġthey": 727, + "Ġthis": 705, + "Ġthree": 902, + "Ġthroug": 923, + "Ġtime": 783, + "Ġto": 391, + "Ġtra": 836, + "Ġtw": 639, + "Ġtwo": 688, + "Ġun": 544, + "Ġup": 898, + "Ġus": 668, + "Ġused": 988, + "Ġv": 495, + "Ġw": 363, + "Ġwas": 422, + "Ġwere": 525, + "Ġwh": 443, + "Ġwhen": 851, + "Ġwhere": 995, + "Ġwhich": 573, + "Ġwhile": 935, + "Ġwho": 724, + "Ġwit": 451, + "Ġwith": 994, + "Ġwith": 466, + "Ġwor": 643, + "Ġwould": 801, + "Ġwrit": 937, + "Ġy": 580, + "Ġyear": 987, + "Ġâ": 556, + "ĠâĢ": 565, + "ĠâĢĵ": 631, + "ĠĊ": 412, + "Ģ": 161, + "Ģ": 223, + "ģ": 162, + "ģ": 273, + "Ĥ": 163, + "Ĥ": 262, + "ĥ": 164, + "ĥ": 337, + "Ħ": 165, + "Ħ": 278, + "ħ": 166, + "ħ": 281, + "Ĩ": 167, + "Ĩ": 308, + "ĩ": 168, + "ĩ": 225, + "Ī": 169, + "Ī": 221, + "ī": 170, + "ī": 244, + "Ĭ": 171, + "Ĭ": 315, + "ĭ": 172, + "ĭ": 321, + "Į": 173, + "Į": 324, + "į": 174, + "į": 302, + "İ": 175, + "İ": 249, + "ı": 176, + "ı": 332, + "IJ": 177, + "IJ": 295, + "ij": 178, + "ij": 313, + "Ĵ": 179, + "Ĵ": 328, + "ĵ": 180, + "ĵ": 312, + "Ķ": 181, + "Ķ": 256, + "ķ": 182, + "ķ": 314, + "ĸ": 183, + "ĸ": 277, + "Ĺ": 184, + "Ĺ": 322, + "ĺ": 185, + "ĺ": 285, + "Ļ": 186, + "Ļ": 267, + "ļ": 187, + "ļ": 290, + "Ľ": 188, + "Ľ": 311, + "ľ": 189, + "ľ": 299, + "Ŀ": 190, + "Ŀ": 291, + "ŀ": 191, + "ŀ": 293, + "Ł": 192, + "Ł": 335, + "ł": 193, + "ł": 252, + "Ń": 194, + "Ń": 297 +} diff --git a/unet/config.json b/unet/config.json index 23816800ebb5473658990794f728b323b5c98c75..e1453ef587c27cbd083e12bedb240155ea25c595 100644 --- a/unet/config.json +++ b/unet/config.json @@ -1,23 +1,54 @@ { - "_class_name": "FlaxUNet2DConditionModel", - "_diffusers_version": "0.10.0.dev0", + "_class_name": "UNet2DConditionModel", + "_diffusers_version": "0.15.0.dev0", + "_name_or_path": "./unet", + "act_fn": "silu", + "addition_embed_type": null, "attention_head_dim": 8, "block_out_channels": [ 32, 64 ], + "center_input_sample": false, + "class_embed_type": null, + "class_embeddings_concat": false, + "conv_in_kernel": 3, + "conv_out_kernel": 3, "cross_attention_dim": 32, + "cross_attention_norm": false, "down_block_types": [ "DownBlock2D", "CrossAttnDownBlock2D" ], + "downsample_padding": 1, "dropout": 0.0, + "dual_cross_attention": false, + "encoder_hid_dim": null, + "flip_sin_to_cos": true, + "freq_shift": 0, "in_channels": 4, "layers_per_block": 2, + "mid_block_scale_factor": 1, + "mid_block_type": "UNetMidBlock2DCrossAttn", + "norm_eps": 1e-05, + "norm_num_groups": 32, + "num_class_embeds": null, + "only_cross_attention": false, "out_channels": 4, + "projection_class_embeddings_input_dim": null, + "resnet_out_scale_factor": 1.0, + "resnet_skip_time_act": false, + "resnet_time_scale_shift": "default", "sample_size": 32, + "time_cond_proj_dim": null, + "time_embedding_act_fn": null, + "time_embedding_dim": null, + "time_embedding_type": "positional", + "timestep_post_act": null, "up_block_types": [ "CrossAttnUpBlock2D", "UpBlock2D" - ] + ], + "upcast_attention": false, + "use_linear_projection": false } diff --git a/unet/diffusion_pytorch_model.fp16.bin b/unet/diffusion_pytorch_model.fp16.bin new file mode 100644 index 0000000000000000000000000000000000000000..745d9ac93f1e09c2c48c392cf365d9fa1ec8e107 --- /dev/null +++ b/unet/diffusion_pytorch_model.fp16.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:56f6ef91ee2b491611f5db5fb39f5e87f9818777d290ee0535f009f665800cec +size 5831675 diff --git a/unet/diffusion_pytorch_model.fp16.safetensors b/unet/diffusion_pytorch_model.fp16.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..719365c8546b5bb6466b3bc01ba5629df636d7eb --- /dev/null +++ b/unet/diffusion_pytorch_model.fp16.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fdd8e3f36cc4ccb30d480171f60190e66add12b160bb3cf3d0807585e323c9c1 +size 5742166 diff --git a/unet/diffusion_pytorch_model.safetensors b/unet/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..719365c8546b5bb6466b3bc01ba5629df636d7eb --- /dev/null +++ b/unet/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:fdd8e3f36cc4ccb30d480171f60190e66add12b160bb3cf3d0807585e323c9c1 +size 5742166 diff --git a/vae/config.json b/vae/config.json index 08790c0388b59d2a88718a67239ce384a1415cbd..e99f7866edf20876871438c38de63896beef31e7 100644 --- a/vae/config.json +++ b/vae/config.json @@ -1,6 +1,7 @@ { - "_class_name": "FlaxAutoencoderKL", - "_diffusers_version": "0.4.0.dev0", + "_class_name": "AutoencoderKL", + "_diffusers_version": "0.15.0.dev0", + "_name_or_path": "./vae", "act_fn": "silu", "block_out_channels": [ 32, @@ -16,6 +17,7 @@ "norm_num_groups": 32, "out_channels": 3, "sample_size": 32, + "scaling_factor": 0.18215, "up_block_types": [ "UpDecoderBlock2D", "UpDecoderBlock2D" diff --git a/vae/diffusion_pytorch_model.fp16.bin b/vae/diffusion_pytorch_model.fp16.bin new file mode 100644 index 0000000000000000000000000000000000000000..503ae2a7c9d9840319d048b146557910924ba1d4 --- /dev/null +++ b/vae/diffusion_pytorch_model.fp16.bin @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6c6d1d95d5b005bc1f82ec4bc56cceacdf14fca0182f56fa2ff1dcb0594980c5 +size 2681375 diff --git a/vae/diffusion_pytorch_model.fp16.safetensors b/vae/diffusion_pytorch_model.fp16.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..f3178383cf7783f180ee1842a6638c38ef2f16fe --- /dev/null +++ b/vae/diffusion_pytorch_model.fp16.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:91bcd25843fd1f2460bc7296f7fc3caafddeae5c19b77157364693c740b5a976 +size 2646544 diff --git a/vae/diffusion_pytorch_model.safetensors b/vae/diffusion_pytorch_model.safetensors new file mode 100644 index 0000000000000000000000000000000000000000..f3178383cf7783f180ee1842a6638c38ef2f16fe --- /dev/null +++ b/vae/diffusion_pytorch_model.safetensors @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:91bcd25843fd1f2460bc7296f7fc3caafddeae5c19b77157364693c740b5a976 +size 2646544