Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -27,8 +27,10 @@ class CustomOVModelVaeDecoder(OVModelVaeDecoder):
|
|
27 |
|
28 |
pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""})
|
29 |
|
30 |
-
|
31 |
-
|
|
|
|
|
32 |
pipe.reshape( batch_size=-1, height=HIGH, width=WIDTH, num_images_per_prompt=1)
|
33 |
#pipe.load_textual_inversion("./badhandv4.pt", "badhandv4")
|
34 |
#pipe.load_textual_inversion("./Konpeto.pt", "Konpeto")
|
|
|
27 |
|
28 |
pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""})
|
29 |
|
30 |
+
#有taesd很醜
|
31 |
+
#taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
|
32 |
+
#pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)
|
33 |
+
|
34 |
pipe.reshape( batch_size=-1, height=HIGH, width=WIDTH, num_images_per_prompt=1)
|
35 |
#pipe.load_textual_inversion("./badhandv4.pt", "badhandv4")
|
36 |
#pipe.load_textual_inversion("./Konpeto.pt", "Konpeto")
|