Upload FluxPipeline

model_index.json

@@ -1,7 +1,15 @@
 {
   "_class_name": "FluxPipeline",
-  "_diffusers_version": "0.31.0",
+  "_diffusers_version": "0.32.1",
   "_name_or_path": "ostris/Flex.1-alpha",
+  "feature_extractor": [
+    null,
+    null
+  ],
+  "image_encoder": [
+    null,
+    null
+  ],
   "scheduler": [
     "diffusers",
     "FlowMatchEulerDiscreteScheduler"

scheduler/scheduler_config.json

@@ -1,11 +1,16 @@
 {
   "_class_name": "FlowMatchEulerDiscreteScheduler",
-  "_diffusers_version": "0.31.0",
+  "_diffusers_version": "0.32.1",
   "base_image_seq_len": 256,
   "base_shift": 0.5,
+  "invert_sigmas": false,
   "max_image_seq_len": 4096,
   "max_shift": 1.15,
   "num_train_timesteps": 1000,
   "shift": 1.0,
-  "use_dynamic_shifting": false
+  "shift_terminal": null,
+  "use_beta_sigmas": false,
+  "use_dynamic_shifting": false,
+  "use_exponential_sigmas": false,
+  "use_karras_sigmas": false
 }

text_encoder/config.json

@@ -19,7 +19,7 @@
   "num_hidden_layers": 12,
   "pad_token_id": 1,
   "projection_dim": 768,
-  "torch_dtype": "float32",
+  "torch_dtype": "float16",
   "transformers_version": "4.47.1",
   "vocab_size": 49408
 }

text_encoder/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1bea3f0b6e1491e0919d379c5fa480d72604e7cc707e097643ba1401673062af
-size 492265168
+oid sha256:f2d7e83d85c915df4f9f5a04c43dc173c279850fe453e3da8891b5941cd67034
+size 246144152

transformer/config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "FluxTransformer2DModel",
-  "_diffusers_version": "0.31.0",
+  "_diffusers_version": "0.32.1",
   "_name_or_path": "ostris/Flex.1-alpha",
   "attention_head_dim": 128,
   "axes_dims_rope": [
@@ -14,6 +14,7 @@
   "num_attention_heads": 24,
   "num_layers": 8,
   "num_single_layers": 38,
+  "out_channels": null,
   "patch_size": 1,
   "pooled_projection_dim": 768,
   "quantization_config": {
@@ -23,7 +24,7 @@
     "bnb_4bit_quant_storage": "uint8",
     "bnb_4bit_quant_type": "nf4",
     "bnb_4bit_use_double_quant": false,
-    "llm_int8_enable_fp32_cpu_offload": true,
+    "llm_int8_enable_fp32_cpu_offload": false,
     "llm_int8_has_fp16_weight": false,
     "llm_int8_skip_modules": null,
     "llm_int8_threshold": 6.0,

transformer/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7728985638a96da9be2bc06e1c7e5ccaadec47929d362616361335d623446a97
-size 4595086819
+oid sha256:d2a7b9834442378f8729c70916d5db08780b53851a75edc50ba732653639df25
+size 4599218587

vae/config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "AutoencoderKL",
-  "_diffusers_version": "0.31.0",
+  "_diffusers_version": "0.32.1",
   "_name_or_path": "/root/.cache/huggingface/hub/models--ostris--Flex.1-alpha/snapshots/5dd0b1bc2a9421b891abcf7c9218993f696dd550/vae",
   "act_fn": "silu",
   "block_out_channels": [

vae/diffusion_pytorch_model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8c717328c8ad41faab2ccfd52ae17332505c6833cf176aad56e7b58f2c4d4c94
-size 335306212
+oid sha256:f58ad0fbb7e0621871b778bf36a6faeb2c46a79adb32b898b60acd1f9560f890
+size 167666654