Yntec commited on
Commit
c5455b9
1 Parent(s): 39b5b7e

Adding `diffusers` weights of this model

Browse files

This PR adds fp32 and fp16 weights in PyTorch and safetensors format to Yntec/SQUEE

model_index.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "StableDiffusionPipeline",
3
+ "_diffusers_version": "0.25.0.dev0",
4
+ "feature_extractor": [
5
+ "transformers",
6
+ "CLIPFeatureExtractor"
7
+ ],
8
+ "image_encoder": [
9
+ null,
10
+ null
11
+ ],
12
+ "requires_safety_checker": true,
13
+ "safety_checker": [
14
+ "stable_diffusion",
15
+ "StableDiffusionSafetyChecker"
16
+ ],
17
+ "scheduler": [
18
+ "diffusers",
19
+ "PNDMScheduler"
20
+ ],
21
+ "text_encoder": [
22
+ "transformers",
23
+ "CLIPTextModel"
24
+ ],
25
+ "tokenizer": [
26
+ "transformers",
27
+ "CLIPTokenizer"
28
+ ],
29
+ "unet": [
30
+ "diffusers",
31
+ "UNet2DConditionModel"
32
+ ],
33
+ "vae": [
34
+ "diffusers",
35
+ "AutoencoderKL"
36
+ ]
37
+ }
safety_checker/config.json ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "CompVis/stable-diffusion-safety-checker",
3
+ "architectures": [
4
+ "StableDiffusionSafetyChecker"
5
+ ],
6
+ "initializer_factor": 1.0,
7
+ "logit_scale_init_value": 2.6592,
8
+ "model_type": "clip",
9
+ "projection_dim": 768,
10
+ "text_config": {
11
+ "dropout": 0.0,
12
+ "hidden_size": 768,
13
+ "intermediate_size": 3072,
14
+ "model_type": "clip_text_model",
15
+ "num_attention_heads": 12
16
+ },
17
+ "torch_dtype": "float16",
18
+ "transformers_version": "4.35.2",
19
+ "vision_config": {
20
+ "dropout": 0.0,
21
+ "hidden_size": 1024,
22
+ "intermediate_size": 4096,
23
+ "model_type": "clip_vision_model",
24
+ "num_attention_heads": 16,
25
+ "num_hidden_layers": 24,
26
+ "patch_size": 14
27
+ }
28
+ }
safety_checker/model.fp16.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57ecdfa243b170f9b4cb3eefaf0f64552ef78fc0bf0eb1c5b9675308447184f6
3
+ size 608016280
safety_checker/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb351a5ded815c3ff744968ad9c6b218d071b9d313d04f35e813b84b4c0ffde8
3
+ size 1215979664
scheduler/scheduler_config.json CHANGED
@@ -1,27 +1,15 @@
1
  {
2
- "_class_name": "DDIMScheduler",
3
- "_diffusers_version": "0.18.0.dev0",
4
-
5
  "beta_end": 0.012,
6
  "beta_schedule": "scaled_linear",
7
  "beta_start": 0.00085,
8
  "clip_sample": false,
9
- "clip_sample_range": 1.0,
10
- "dynamic_thresholding_ratio": 0.995,
11
-
12
-
13
  "num_train_timesteps": 1000,
14
  "prediction_type": "epsilon",
15
- "rescale_betas_zero_snr": false,
16
- "sample_max_value": 1.0,
17
  "set_alpha_to_one": false,
18
-
19
-
20
  "steps_offset": 1,
21
- "thresholding": false,
22
  "timestep_spacing": "leading",
23
  "trained_betas": null
24
-
25
-
26
-
27
- }
 
1
  {
2
+ "_class_name": "PNDMScheduler",
3
+ "_diffusers_version": "0.25.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
  "timestep_spacing": "leading",
14
  "trained_betas": null
15
+ }
 
 
 
unet/diffusion_pytorch_model.fp16.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5eb162fac97689dc107ebe615d8be59e88193507d28903de4294e45a14a54d8e
3
  size 1719125304
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:374142504757704174d840a34eedade01dbed74d85b89affdad6b9b590a285d6
3
  size 1719125304
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1bd921ef63a276cfaed9d0c8830d9504e9f812a043ac81ac48626dd42544ff7c
3
  size 3438167536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:19b81f875bbdc4c61f885e72cefd41796466860b574d8e161e54302933dc0008
3
  size 3438167536