Spaces:
Runtime error
Runtime error
Update app.py
Browse filesChange that was reflected in README.md of model card
app.py
CHANGED
@@ -39,7 +39,7 @@ height = int(os.getenv("IMAGE_HEIGHT", "512"))
|
|
39 |
num_images = int(os.getenv("NUM_IMAGES", "1"))
|
40 |
|
41 |
scheduler = LCMScheduler.from_pretrained(model_id, subfolder="scheduler")
|
42 |
-
pipe = OVLatentConsistencyModelPipeline.from_pretrained(model_id, scheduler = scheduler, compile = False)
|
43 |
pipe.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
|
44 |
pipe.compile()
|
45 |
|
|
|
39 |
num_images = int(os.getenv("NUM_IMAGES", "1"))
|
40 |
|
41 |
scheduler = LCMScheduler.from_pretrained(model_id, subfolder="scheduler")
|
42 |
+
pipe = OVLatentConsistencyModelPipeline.from_pretrained(model_id, scheduler = scheduler, compile = False, ov_config = {"CACHE_DIR":""})
|
43 |
pipe.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
|
44 |
pipe.compile()
|
45 |
|