Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from huggingface_hub import snapshot_download
|
|
8 |
import openvino.runtime as ov
|
9 |
from typing import Optional, Dict
|
10 |
|
11 |
-
model_id = "Disty0/
|
12 |
#model_id = "Disty0/sotediffusion-v2" #不可
|
13 |
|
14 |
#1024*512 記憶體不足
|
@@ -29,12 +29,12 @@ pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_c
|
|
29 |
taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
|
30 |
pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)
|
31 |
pipe.reshape( batch_size=-1, height=HIGH, width=WIDTH, num_images_per_prompt=1)
|
32 |
-
pipe.load_textual_inversion("./badhandv4.pt", "badhandv4")
|
33 |
#pipe.load_textual_inversion("./Konpeto.pt", "Konpeto")
|
34 |
#<shigure-ui-style>
|
35 |
-
pipe.load_textual_inversion("sd-concepts-library/shigure-ui-style")
|
36 |
-
pipe.load_textual_inversion("sd-concepts-library/ruan-jia")
|
37 |
-
pipe.load_textual_inversion("sd-concepts-library/agm-style-nao")
|
38 |
|
39 |
|
40 |
pipe.compile()
|
|
|
8 |
import openvino.runtime as ov
|
9 |
from typing import Optional, Dict
|
10 |
|
11 |
+
model_id = "Disty0/LCM_SoteMix"
|
12 |
#model_id = "Disty0/sotediffusion-v2" #不可
|
13 |
|
14 |
#1024*512 記憶體不足
|
|
|
29 |
taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
|
30 |
pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)
|
31 |
pipe.reshape( batch_size=-1, height=HIGH, width=WIDTH, num_images_per_prompt=1)
|
32 |
+
#pipe.load_textual_inversion("./badhandv4.pt", "badhandv4")
|
33 |
#pipe.load_textual_inversion("./Konpeto.pt", "Konpeto")
|
34 |
#<shigure-ui-style>
|
35 |
+
#pipe.load_textual_inversion("sd-concepts-library/shigure-ui-style")
|
36 |
+
#pipe.load_textual_inversion("sd-concepts-library/ruan-jia")
|
37 |
+
#pipe.load_textual_inversion("sd-concepts-library/agm-style-nao")
|
38 |
|
39 |
|
40 |
pipe.compile()
|