deinferno commited on
Commit
bf319ef
·
1 Parent(s): 7b8f80d

Update app.py

Browse files

Change that was reflected in README.md of model card

Files changed (1) hide show
  1. app.py +1 -1
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