Run pipeline
Browse files
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 |
-
|
| 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(
|