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 |
|