Commit 216ad50
committed
fix(models): check for all expected ONNX files during conversion
CodeT5 produces multiple ONNX files (encoder_model.onnx and decoder_model.onnx)
but convertToOnnx was only checking for a single model.onnx file. This caused
the conversion step to be marked as complete when the files didn't actually exist,
leading to build failures in the quantization step.
Fix: Use config.files to check for all expected ONNX files based on model type.1 parent 68641fe commit 216ad50
1 file changed
+5
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
156 | 156 | | |
157 | 157 | | |
158 | 158 | | |
159 | | - | |
160 | 159 | | |
161 | | - | |
| 160 | + | |
| 161 | + | |
| 162 | + | |
| 163 | + | |
| 164 | + | |
162 | 165 | | |
163 | 166 | | |
164 | 167 | | |
| |||
0 commit comments