Bethie commited on
Commit
74eeb7f
·
verified ·
1 Parent(s): 78c0915

Run pipeline

Browse files
Files changed (1) hide show
  1. code_inference/run_controlnext.py +1 -2
code_inference/run_controlnext.py CHANGED
@@ -260,8 +260,7 @@ if __name__ == "__main__":
260
  scheduler = DDPMScheduler.from_pretrained(SCHEDULER_PATH)
261
 
262
  controlnet = ort.InferenceSession(CONTROLNEXT_ONNX_PATH, providers=providers, sess_options=session_options)
263
- #image_encoder = ort.InferenceSession(IMAGE_ENCODER_ONNX_PATH, providers=providers, provider_options=provider_options_0)
264
- image_encoder = CLIPVisionModelWithProjection.from_pretrained('h94/IP-Adapter', subfolder = 'sdxl_models/image_encoder').to(device, dtype=torch.float32)
265
  image_proj = ort.InferenceSession(PROJ_ONNX_PATH, providers=providers, sess_options=session_options)
266
 
267
  log_validation(
 
260
  scheduler = DDPMScheduler.from_pretrained(SCHEDULER_PATH)
261
 
262
  controlnet = ort.InferenceSession(CONTROLNEXT_ONNX_PATH, providers=providers, sess_options=session_options)
263
+ image_encoder = ort.InferenceSession(IMAGE_ENCODER_ONNX_PATH, providers=providers, provider_options=provider_options_1)
 
264
  image_proj = ort.InferenceSession(PROJ_ONNX_PATH, providers=providers, sess_options=session_options)
265
 
266
  log_validation(