optimum-neuron-cache / inference-cache-config /stable-diffusion-v1-5.json
Mattral's picture
Duplicate from aws-neuron/optimum-neuron-cache
5594fd0
Raw
History Blame Contribute Delete
377 Bytes
{
"jyoung105/stable-diffusion-v1-5": [
{ "task": "stable-diffusion", "batch_size": 1, "height": 512, "width": 512, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" },
{ "task": "stable-diffusion", "batch_size": 1, "height": 768, "width": 768, "num_images_per_prompt": 1, "auto_cast": "matmul", "auto_cast_type": "bf16" }
]
}