Commit ·
90606bb
1
Parent(s): 6e48cc8
Update README.md
Browse files
README.md
CHANGED
|
@@ -6,12 +6,13 @@ base_model:
|
|
| 6 |
|
| 7 |
|
| 8 |
Experimental ONNX conversion of kaloscope model.
|
| 9 |
-
These need a re-export because i didnt specify dynamic axes during export which probably locked the model behind batch size = 1; otherwise they seem to work perfectly fine
|
|
|
|
| 10 |
|
| 11 |
[🤗 Space Demo](https://huggingface.co/spaces/DraconicDragon/Kaloscope-artist-style-classifier) for ONNX & PyTorch inference implementation (incl. timm+lsnet; OpenVINO accelerate CPU inference; no Triton required - refer to [ska.py](https://huggingface.co/spaces/DraconicDragon/Kaloscope-artist-style-classifier/blob/main/lsnet/ska.py) or [here](https://github.com/spawner1145/comfyui-lsnet/pull/2))
|
| 12 |
Barebones standalone ONNX inference script (no timm or lsnet; scores are a tiny bit different - probably different img preprocessing): [onnx_barebones_inference.py](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/onnx_barebones_inference.py)
|
| 13 |
|
| 14 |
-
- `kaloscope_1-0.onnx`: Exported from `best_checkpoint.pth` original Kaloscope release | dynamo=
|
| 15 |
- `kaloscope_1-1.onnx`: Exported from `224-85.65/best_checkpoint.pth` aka v1.1 | Same settings as v1.0
|
| 16 |
|
| 17 |
Conversion/Export script can be found in convert-scripts folder or:
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
Experimental ONNX conversion of kaloscope model.
|
| 9 |
+
~~These need a re-export because i didnt specify dynamic axes during export which probably locked the model behind batch size = 1; otherwise they seem to work perfectly fine~~
|
| 10 |
+
25th oct: replaced existing onnx files with new exports that use dynamo=False and `dynamic_axes={"input": {0: "batch_size"}, "output": {0: "batch_size"},}` - scores seems to be same as original pytorch model and dynamo=True export
|
| 11 |
|
| 12 |
[🤗 Space Demo](https://huggingface.co/spaces/DraconicDragon/Kaloscope-artist-style-classifier) for ONNX & PyTorch inference implementation (incl. timm+lsnet; OpenVINO accelerate CPU inference; no Triton required - refer to [ska.py](https://huggingface.co/spaces/DraconicDragon/Kaloscope-artist-style-classifier/blob/main/lsnet/ska.py) or [here](https://github.com/spawner1145/comfyui-lsnet/pull/2))
|
| 13 |
Barebones standalone ONNX inference script (no timm or lsnet; scores are a tiny bit different - probably different img preprocessing): [onnx_barebones_inference.py](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/onnx_barebones_inference.py)
|
| 14 |
|
| 15 |
+
- `kaloscope_1-0.onnx`: Exported from `best_checkpoint.pth` original Kaloscope release | dynamo=False, dynamic_axes={"input": {0: "batch_size"}, "output": {0: "batch_size"},}, opset_version=None (torch 2.8.0 used here defaults to 18 when None), optimization/constant folding enabled
|
| 16 |
- `kaloscope_1-1.onnx`: Exported from `224-85.65/best_checkpoint.pth` aka v1.1 | Same settings as v1.0
|
| 17 |
|
| 18 |
Conversion/Export script can be found in convert-scripts folder or:
|