HelloSun commited on
Commit
9be9826
·
verified ·
1 Parent(s): 81e5760

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -7,7 +7,6 @@ import torch
7
  from huggingface_hub import snapshot_download
8
  import openvino.runtime as ov
9
  from typing import Optional, Dict
10
- from transformers import CLIPFeatureExtractor, CLIPTokenizer
11
 
12
 
13
 
@@ -18,8 +17,6 @@ model_id = "Disty0/LCM_SoteMix"
18
  HIGH=768
19
  WIDTH=512
20
 
21
- tokenizer = CLIPTokenizer.from_pretrained(model_id)
22
-
23
  batch_size = -1
24
  class CustomOVModelVaeDecoder(OVModelVaeDecoder):
25
  def __init__(
@@ -28,7 +25,7 @@ class CustomOVModelVaeDecoder(OVModelVaeDecoder):
28
  super(OVModelVaeDecoder, self).__init__(model, parent_model, ov_config, "vae_decoder", model_dir)
29
 
30
 
31
- pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""}, tokenizer=tokenizer,)
32
 
33
  taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
34
  pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)
 
7
  from huggingface_hub import snapshot_download
8
  import openvino.runtime as ov
9
  from typing import Optional, Dict
 
10
 
11
 
12
 
 
17
  HIGH=768
18
  WIDTH=512
19
 
 
 
20
  batch_size = -1
21
  class CustomOVModelVaeDecoder(OVModelVaeDecoder):
22
  def __init__(
 
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":""}, )
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)