HelloSun commited on
Commit
d5bd739
1 Parent(s): 18b3b5c

Update app.py

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