Commit
·
39fe6ac
1
Parent(s):
5255f7f
load lora
Browse files- handler.py +3 -3
- payload.json +5 -0
handler.py
CHANGED
|
@@ -78,16 +78,16 @@ class EndpointHandler():
|
|
| 78 |
self.pipe.scheduler.config
|
| 79 |
)
|
| 80 |
|
| 81 |
-
# load
|
| 82 |
self.pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
|
| 83 |
-
self.pipe.
|
| 84 |
|
| 85 |
self.pipe.cuda()
|
| 86 |
self.pipe.load_ip_adapter_instantid(face_adapter)
|
| 87 |
self.pipe.image_proj_model.to("cuda")
|
| 88 |
self.pipe.unet.to("cuda")
|
|
|
|
| 89 |
self.pipe.scheduler = diffusers.LCMScheduler.from_config(self.pipe.scheduler.config)
|
| 90 |
-
self.pipe.enable_lora()
|
| 91 |
|
| 92 |
|
| 93 |
controlnet_pose_model = "thibaud/controlnet-openpose-sdxl-1.0"
|
|
|
|
| 78 |
self.pipe.scheduler.config
|
| 79 |
)
|
| 80 |
|
| 81 |
+
# load LCM
|
| 82 |
self.pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
|
| 83 |
+
self.pipe.enable_lora()
|
| 84 |
|
| 85 |
self.pipe.cuda()
|
| 86 |
self.pipe.load_ip_adapter_instantid(face_adapter)
|
| 87 |
self.pipe.image_proj_model.to("cuda")
|
| 88 |
self.pipe.unet.to("cuda")
|
| 89 |
+
|
| 90 |
self.pipe.scheduler = diffusers.LCMScheduler.from_config(self.pipe.scheduler.config)
|
|
|
|
| 91 |
|
| 92 |
|
| 93 |
controlnet_pose_model = "thibaud/controlnet-openpose-sdxl-1.0"
|
payload.json
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"inputs": "a man",
|
| 3 |
+
"guidance_scale": 5.0,
|
| 4 |
+
"num_inference_steps": 20
|
| 5 |
+
}
|