Update app.py
Browse files
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 =
|
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 |
|