Hugging Face's logo

badaoui
/
black-forest-labs-FLUX.1-dev-neuron

neuron
optimized
aws-neuron
flux
Model card Files Community
badaoui HF Staff commited on
Commit
3a7bf60
·
verified ·
1 Parent(s): e89031f

Add Neuron-optimized files for black-forest-labs/FLUX.1-dev

Browse files

🤖 Neuron Export Bot: Adding AWS Neuron-optimized model files.

Original model: [black-forest-labs/FLUX.1-dev](https://huggingface.co/black-forest-labs/FLUX.1-dev)
Task: flux
Generated by: [badaoui](https://huggingface.co/badaoui)
Generated using: [Optimum Neuron Compiler Space](https://huggingface.co/spaces/optimum/neuron-export)

These files have been pre-compiled for AWS Neuron devices (Inferentia/Trainium) and should provide improved inference performance.

Files changed (13) hide show
  1. model_index.json +1 -1
  2. scheduler/scheduler_config.json +1 -1
  3. text_encoder/model.neuron +1 -1
  4. text_encoder_2/model.neuron/tp_0.pt +1 -1
  5. text_encoder_2/model.neuron/tp_1.pt +2 -2
  6. text_encoder_2/model.neuron/tp_2.pt +2 -2
  7. text_encoder_2/model.neuron/tp_3.pt +2 -2
  8. transformer/config.json +3 -3
  9. transformer/model.neuron +1 -1
  10. vae_decoder/config.json +1 -1
  11. vae_decoder/model.neuron +1 -1
  12. vae_encoder/config.json +1 -1
  13. vae_encoder/model.neuron +1 -1
model_index.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "FluxPipeline",
3
- "_diffusers_version": "0.34.0",
4
  "_name_or_path": "black-forest-labs/FLUX.1-dev",
5
  "feature_extractor": [
6
  null,
 
1
  {
2
  "_class_name": "FluxPipeline",
3
+ "_diffusers_version": "0.35.1",
4
  "_name_or_path": "black-forest-labs/FLUX.1-dev",
5
  "feature_extractor": [
6
  null,
scheduler/scheduler_config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "FlowMatchEulerDiscreteScheduler",
3
- "_diffusers_version": "0.34.0",
4
  "base_image_seq_len": 256,
5
  "base_shift": 0.5,
6
  "invert_sigmas": false,
 
1
  {
2
  "_class_name": "FlowMatchEulerDiscreteScheduler",
3
+ "_diffusers_version": "0.35.1",
4
  "base_image_seq_len": 256,
5
  "base_shift": 0.5,
6
  "invert_sigmas": false,
text_encoder/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:1c69e6e558f4a966e0d296264403ea3ff5c38307f07d2cf1a38410c6c19a0e18
3
  size 307199283
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3512591dd6cb48a23767ca275a6b1f079fcb5f8b74bd4d8f26cd50c6ce1e933
3
  size 307199283
text_encoder_2/model.neuron/tp_0.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:172daf055431a1b8d76aa7e31ff181360e7d1b8929b7e1d13e62a761fc301198
3
  size 2638601250
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a3780b53b407137e183922ca6410e143e425e61d7830809804c06abb9d76be4
3
  size 2638601250
text_encoder_2/model.neuron/tp_1.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3ce2abe965b1464f397e05e8f62ce968bcface0b8452baa337348e7fa5f64acb
3
- size 2638600234
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dd25cf9a28241c599672d52a35c1bb35b50245323a604450180a24c7cc29fd58
3
+ size 2638601246
text_encoder_2/model.neuron/tp_2.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3dc2effa3656bf5115503027c080168ea396a8a2d8688551983482eb5eeec0fd
3
- size 2638601246
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7226fd473978e770c5b12923f08aae6344d97545d03726e9fca711be1dbff929
3
+ size 2638600234
text_encoder_2/model.neuron/tp_3.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7e3e5f966cc268cc087d89c9593c162c6090d3e089745974ce2b7b0d3d56242e
3
- size 2638601316
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f36b9a55103b74b593547ba7d51508beb158f72bccc75a4738208dd8761e8bea
3
+ size 2638601252
transformer/config.json CHANGED
@@ -1,10 +1,10 @@
1
  {
2
  "_class_name": "FluxTransformer2DModel",
3
  "_commit_hash": null,
4
- "_diffusers_version": "0.34.0",
5
  "_use_default_values": [
6
- "out_channels",
7
- "axes_dims_rope"
8
  ],
9
  "attention_head_dim": 128,
10
  "axes_dims_rope": [
 
1
  {
2
  "_class_name": "FluxTransformer2DModel",
3
  "_commit_hash": null,
4
+ "_diffusers_version": "0.35.1",
5
  "_use_default_values": [
6
+ "axes_dims_rope",
7
+ "out_channels"
8
  ],
9
  "attention_head_dim": 128,
10
  "axes_dims_rope": [
transformer/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:71fd430170c7153bf693a80606f339d20e6391d4abcfe7ebf83549dfc179c93f
3
  size 15296732
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34c9dd7506b145bea29a61f5e9924e60b9d8c9de649e4ed95755a0f59a4a897a
3
  size 15296732
vae_decoder/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_commit_hash": null,
4
- "_diffusers_version": "0.34.0",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_commit_hash": null,
4
+ "_diffusers_version": "0.35.1",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
vae_decoder/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7b657b819d09f2bfba1a1c98b0e2692bb704a232775d1e3a672c8539dd2f8879
3
  size 373356467
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b8b59b68bbb662f657a182ce0746362b6b41d0f91ede9767bee1cf5c79de74f0
3
  size 373356467
vae_encoder/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_commit_hash": null,
4
- "_diffusers_version": "0.34.0",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
 
1
  {
2
  "_class_name": "AutoencoderKL",
3
  "_commit_hash": null,
4
+ "_diffusers_version": "0.35.1",
5
  "act_fn": "silu",
6
  "block_out_channels": [
7
  128,
vae_encoder/model.neuron CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cec23b7bf6c9b6b4f502068d834773acef5d2e0f48f9be4db81c70febf8dd3ae
3
  size 223240499
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8725df04416358d93483724b6c1e2999df059b5383d66bc8c4a8bc9e162b5d32
3
  size 223240499