HelloSun commited on
Commit
e698613
·
verified ·
1 Parent(s): 6694cc8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ model_id = "yujiepan/dreamshaper-8-lcm-openvino-w8a8"
19
 
20
  #pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, safety_checker=safety_checker)
21
  #pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, device='CPU',)
22
- pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, device='CPU',)
23
  #batch_size, num_images, height, width = 1, 1, 512, 512
24
  #pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
25
 
 
19
 
20
  #pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, safety_checker=safety_checker)
21
  #pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, device='CPU',)
22
+ pipeline = OVStableDiffusionPipeline.from_pretrained(model_id, device='CPU',)
23
  #batch_size, num_images, height, width = 1, 1, 512, 512
24
  #pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
25