HelloSun commited on
Commit
ad687bc
·
verified ·
1 Parent(s): a03062d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -25,8 +25,9 @@ pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_c
25
 
26
  taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
27
  pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)
28
-
29
  pipe.reshape( batch_size=-1, height=HIGH, width=WIDTH, num_images_per_prompt=1)
 
 
30
  pipe.compile()
31
 
32
  prompt=""
 
25
 
26
  taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
27
  pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)
 
28
  pipe.reshape( batch_size=-1, height=HIGH, width=WIDTH, num_images_per_prompt=1)
29
+ pipe.load_textual_inversion("./badhandv4.pt", "badhandv4")
30
+ pipe.load_textual_inversion("./Konpeto.pt", "Konpeto")
31
  pipe.compile()
32
 
33
  prompt=""