Update app.py

app.py

@@ -7,6 +7,9 @@ import torch
 from huggingface_hub import snapshot_download
 import openvino.runtime as ov
 from typing import Optional, Dict
+from transformers import CLIPFeatureExtractor, CLIPTokenizer
+
+
 
 model_id = "Disty0/LCM_SoteMix"
 #model_id = "Disty0/sotediffusion-v2" #不可
@@ -15,6 +18,7 @@ model_id = "Disty0/LCM_SoteMix"
 HIGH=768
 WIDTH=512
 
+tokenizer = CLIPTokenizer.from_pretrained(model_id)
 
 batch_size = -1
 class CustomOVModelVaeDecoder(OVModelVaeDecoder):
@@ -24,7 +28,7 @@ class CustomOVModelVaeDecoder(OVModelVaeDecoder):
         super(OVModelVaeDecoder, self).__init__(model, parent_model, ov_config, "vae_decoder", model_dir)
 
 
-pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""})
+pipe = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""}, tokenizer=tokenizer,)
 
 taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
 pipe.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipe, model_dir = taesd_dir)