DraconicDragon commited on
Commit
df535f0
·
verified ·
1 Parent(s): 8521211

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -3
README.md CHANGED
@@ -5,15 +5,14 @@ base_model:
5
  ---
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: [colab notebook](https://huggingface.co/DraconicDragon/Kaloscope-onnx/blob/main/convert_scripts/kaloscope_pth2onnx_colab.ipynb)
19
  theres an extra pytorch ema only version (`best_checkpoint_ema.pth`) of the model which is roughly 4 times smaller but seems to output different results; did not do extensive tests. ONNX version recommended over this (script used: [save_ema.py](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/save_ema.py))
 
5
  ---
6
 
7
 
8
+ ONNX conversion of kaloscope model.
 
 
9
 
10
  [🤗 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))
11
  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)
12
 
13
  - `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
14
  - `kaloscope_1-1.onnx`: Exported from `224-85.65/best_checkpoint.pth` aka v1.1 | Same settings as v1.0
15
+ - `v2.0/kaloscope_2-0.onnx`: Exported from `Kaloscope2.0/448-90.13/best_checkpoint.pth` aka v2.0 | Same settings as v1.0
16
 
17
  Conversion/Export script: [colab notebook](https://huggingface.co/DraconicDragon/Kaloscope-onnx/blob/main/convert_scripts/kaloscope_pth2onnx_colab.ipynb)
18
  theres an extra pytorch ema only version (`best_checkpoint_ema.pth`) of the model which is roughly 4 times smaller but seems to output different results; did not do extensive tests. ONNX version recommended over this (script used: [save_ema.py](https://huggingface.co/DraconicDragon/Kaloscope-onnx-ema/blob/main/save_ema.py))