Hugging Face's logo

aws-neuron
/
optimum-neuron-cache

Model card Files
xet
Community
603
Jingya HF Staff commited on
Commit
e5f06c7
·
verified ·
1 Parent(s): f5e49e5

add pixart and remove deprecated

Browse files
Files changed (1) hide show
  1. inference-cache-config/{stable-diffusion.json → diffusion.json} +12 -6
inference-cache-config/{stable-diffusion.json → diffusion.json} RENAMED
@@ -8,15 +8,21 @@
8
  { "task": "stable-diffusion", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
9
  ],
10
  "stabilityai/stable-diffusion-xl-base-1.0": [
11
- { "task": "stable-diffusion-xl", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" },
12
- { "task": "stable-diffusion-xl", "batch_size": 1, "height": 1024, "width": 1024, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
13
  ],
14
  "stabilityai/stable-diffusion-xl-refiner-1.0": [
15
- { "task": "stable-diffusion-xl", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" },
16
- { "task": "stable-diffusion-xl", "batch_size": 1, "height": 1024, "width": 1024, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
17
  ],
18
  "stabilityai/sdxl-turbo": [
19
- { "task": "stable-diffusion-xl", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" },
20
- { "task": "stable-diffusion-xl", "batch_size": 1, "height": 1024, "width": 1024, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
 
 
 
 
 
 
21
  ]
22
  }
 
8
  { "task": "stable-diffusion", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
9
  ],
10
  "stabilityai/stable-diffusion-xl-base-1.0": [
11
+ { "task": "stable-diffusion", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" },
12
+ { "task": "stable-diffusion", "batch_size": 1, "height": 1024, "width": 1024, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
13
  ],
14
  "stabilityai/stable-diffusion-xl-refiner-1.0": [
15
+ { "task": "stable-diffusion", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" },
16
+ { "task": "stable-diffusion", "batch_size": 1, "height": 1024, "width": 1024, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
17
  ],
18
  "stabilityai/sdxl-turbo": [
19
+ { "task": "stable-diffusion", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" },
20
+ { "task": "stable-diffusion", "batch_size": 1, "height": 1024, "width": 1024, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
21
+ ],
22
+ "PixArt-alpha/PixArt-XL-2-512x512": [
23
+ { "task": "pixart", "batch_size": 1, "height": 512, "width": 512, "num_images_per_prompt": 1, "torch_dtype": "bfloat16", "sequence_length": 120 }
24
+ ],
25
+ "Jingya/pixart_sigma_pipe_xl_2_512_ms": [
26
+ { "task": "pixart", "batch_size": 1, "height": 512, "width": 512, "num_images_per_prompt": 1, "torch_dtype": "bfloat16", "sequence_length": 120 }
27
  ]
28
  }