Update app.py
Browse files
app.py
CHANGED
@@ -25,7 +25,7 @@ class CustomOVModelVaeDecoder(OVModelVaeDecoder):
|
|
25 |
super(OVModelVaeDecoder, self).__init__(model, parent_model, ov_config, "vae_decoder", model_dir)
|
26 |
|
27 |
|
28 |
-
pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""}, torch_dtype=torch.
|
29 |
|
30 |
taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
|
31 |
pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)
|
|
|
25 |
super(OVModelVaeDecoder, self).__init__(model, parent_model, ov_config, "vae_decoder", model_dir)
|
26 |
|
27 |
|
28 |
+
pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""}, torch_dtype=torch.bool, )
|
29 |
|
30 |
taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
|
31 |
pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)
|