Spaces:

HelloSun
/

LCM_Dreamshaper_v7-int8-ov

Sleeping

HelloSun commited on Jul 30, 2024

Commit

fe0774e

verified ·

1 Parent(s): 3d65110

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ from diffusers import DiffusionPipeline
 model_id = "OpenVINO/LCM_Dreamshaper_v7-int8-ov"
 #pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, safety_checker=safety_checker)
 pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False)
-pipeline.load_lora_weights("EvilEngine/easynegative")
 batch_size, num_images, height, width = 1, 1, 512, 512
 pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)

 model_id = "OpenVINO/LCM_Dreamshaper_v7-int8-ov"
 #pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, safety_checker=safety_checker)
 pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False)
+pipeline.load_tf_weights("EvilEngine/easynegative")
 batch_size, num_images, height, width = 1, 1, 512, 512
 pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)