yamildiego commited on
Commit
39fe6ac
·
1 Parent(s): 5255f7f
Files changed (2) hide show
  1. handler.py +3 -3
  2. payload.json +5 -0
handler.py CHANGED
@@ -78,16 +78,16 @@ class EndpointHandler():
78
  self.pipe.scheduler.config
79
  )
80
 
81
- # load and disable LCM
82
  self.pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl")
83
- self.pipe.disable_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
  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
+ }