Define path to ONNX weights
Browse files
code_inference/configs.py
CHANGED
|
@@ -22,7 +22,7 @@ SCHEDULER_PATH = 'onnx_1/scheduler'
|
|
| 22 |
|
| 23 |
CONTROLNEXT_ONNX_PATH = 'onnx_1/controlnext/model.onnx'
|
| 24 |
|
| 25 |
-
|
| 26 |
|
| 27 |
PROJ_ONNX_PATH = 'onnx_1/proj/model.onnx'
|
| 28 |
|
|
|
|
| 22 |
|
| 23 |
CONTROLNEXT_ONNX_PATH = 'onnx_1/controlnext/model.onnx'
|
| 24 |
|
| 25 |
+
IMAGE_ENCODER_ONNX_PATH = 'onnx_1/image_encoder/model.onnx'
|
| 26 |
|
| 27 |
PROJ_ONNX_PATH = 'onnx_1/proj/model.onnx'
|
| 28 |
|