HelloSun commited on
Commit
18e198c
1 Parent(s): c35386c

Update app.py

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