HelloSun commited on
Commit
3c6891b
·
verified ·
1 Parent(s): 5063a6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -16
app.py CHANGED
@@ -19,22 +19,7 @@ model_id = "OpenVINO/LCM_Dreamshaper_v7-int8-ov"
19
  #pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, safety_checker=safety_checker)
20
  pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False)
21
 
22
- batch_size, num_images, height, width = -1, 1, 1024, 512
23
-
24
-
25
- class CustomOVModelVaeDecoder(OVModelVaeDecoder):
26
- def __init__(
27
- self, model: ov.Model, parent_model: OVBaseModel, ov_config: Optional[Dict[str, str]] = None, model_dir: str = None,
28
- ):
29
- super(OVModelVaeDecoder, self).__init__(model, parent_model, ov_config, "vae_decoder", model_dir)
30
-
31
-
32
- pipeline = OVStableDiffusionPipeline.from_pretrained(model_id, compile = False, ov_config = {"CACHE_DIR":""})
33
-
34
- taesd_dir = snapshot_download(repo_id="deinferno/taesd-openvino")
35
- pipeline.vae_decoder = CustomOVModelVaeDecoder(model = OVBaseModel.load_model(f"{taesd_dir}/vae_decoder/openvino_model.xml"), parent_model = pipeline, model_dir = taesd_dir)
36
-
37
-
38
  pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
39
 
40
  #不可用lora
 
19
  #pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, safety_checker=safety_checker)
20
  pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False)
21
 
22
+ batch_size, num_images, height, width = 1, 1, 1024, 512
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
23
  pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
24
 
25
  #不可用lora