forked from ZillaRU/SD-lcm-tpu
-
Notifications
You must be signed in to change notification settings - Fork 0
/
model_path.py
68 lines (66 loc) · 1.92 KB
/
model_path.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
'''example
"awportrait": {
"name": "awportrait",
"encoder": "te_f32.bmodel",
"unet": "unet_2_1684x_F16.bmodel",
"vae_decoder": "vae_decoder_multize.bmodel",
"vae_encoder": "vae_encoder_multize.bmodel",
"controlnet": "canny_multize",
"latent_shape": {
"512x512": True,
"768x512": True,
"512x768": False
}
},
'''
model_path = {
"hello": {
"name": "hellonijicute",
"encoder": "sdv15_text.bmodel",
"unet": "sdv15_unet_multisize.bmodel",
"vae_decoder": "sdv15_vd_multisize.bmodel",
"vae_encoder": "sdv15_ve_multisize.bmodel",
"latent_shape": {
"512x512": "True",
"768x512": "False",
"512x768": "False"
}
},
"awportrait": {
"name": "awportrait",
"encoder": "te_f32.bmodel",
"unet": "unet_2_1684x_F16.bmodel",
"vae_decoder": "vae_decoder_multize.bmodel",
"vae_encoder": "vae_encoder_multize.bmodel",
"latent_shape": {
"512x512": "True",
"768x512": "False",
"512x768": "False"
}
},
"meinamix":{
"name": "meinamix",
"encoder": "sdv15_te.bmodel",
"unet": "unet_1_1684x_F16.bmodel",
"vae_decoder": "vae_decoder_multize.bmodel",
"vae_encoder": "vae_encoder_multize.bmodel",
"controlnet": "canny_multize",
"latent_shape": {
"512x512": "True",
"768x512": "False",
"512x768": "False"
}
},
"toonyou":{
"name": "toonyou",
"encoder": "sdv15_text.bmodel",
"unet": "sdv15_unet_multisize.bmodel",
"vae_decoder": "sdv15_vd_multisize.bmodel",
"vae_encoder": "sdv15_ve_multisize.bmodel",
"latent_shape": {
"512x512": "True",
"768x512": "True",
"512x768": "True"
}
},
}