Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ model_id = "OpenVINO/LCM_Dreamshaper_v7-int8-ov"
|
|
15 |
#pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, safety_checker=safety_checker)
|
16 |
pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False)
|
17 |
|
18 |
-
batch_size, num_images, height, width = 1, 1, 1024,
|
19 |
pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
|
20 |
pipeline.load_textual_inversion("./badhandv4.pt", "badhandv4")
|
21 |
pipeline.compile()
|
|
|
15 |
#pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, safety_checker=safety_checker)
|
16 |
pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False)
|
17 |
|
18 |
+
batch_size, num_images, height, width = 1, 1, 1024, 512
|
19 |
pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
|
20 |
pipeline.load_textual_inversion("./badhandv4.pt", "badhandv4")
|
21 |
pipeline.compile()
|