Update app.py
Browse files
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=""
|