Jingya HF Staff commited on
Commit
4a034bb
·
verified ·
1 Parent(s): 37da0a7

Rename inference-cache-config/diffusion.json to inference-cache-config/stable-diffusion-v1-5.json

Browse files
inference-cache-config/diffusion.json DELETED
@@ -1,28 +0,0 @@
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" }
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
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
inference-cache-config/stable-diffusion-v1-5.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
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
+ }