HelloSun commited on
Commit
47636ca
·
verified ·
1 Parent(s): 46b3e30

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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/SoteMixV3"
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()