Hugging Face's logo

aws-neuron
/
optimum-neuron-cache

Model card Files Community
235
Jingya HF staff commited on
Commit
5272eb2
•
1 Parent(s): 43fb274

Update inference-cache-config/stable-diffusion.json

Browse files
Files changed (1) hide show
  1. inference-cache-config/stable-diffusion.json +4 -0
inference-cache-config/stable-diffusion.json CHANGED
@@ -1,4 +1,8 @@
1
  {
 
 
 
 
2
  "stabilityai/stable-diffusion-2-1": [
3
  { "task": "stable-diffusion", "batch_size": 1, "height": 512, "width": 512, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" },
4
  { "task": "stable-diffusion", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
 
1
  {
2
+ "jyoung105/stable-diffusion-v1-5": [
3
+ { "task": "stable-diffusion", "batch_size": 1, "height": 512, "width": 512, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" },
4
+ { "task": "stable-diffusion", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
5
+ ],
6
  "stabilityai/stable-diffusion-2-1": [
7
  { "task": "stable-diffusion", "batch_size": 1, "height": 512, "width": 512, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" },
8
  { "task": "stable-diffusion", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }