This commit is contained in:
2025-06-10 18:10:41 -04:00
commit 914de08b49
4 changed files with 208 additions and 0 deletions

14
export_onnx_v2.py Normal file
View File

@@ -0,0 +1,14 @@
model_name = 'ViT-L-16-SigLIP2-512__webli'
from pathlib import Path
name, _, pretrained = model_name.partition("__")
config = OpenCLIPModelConfig(name, pretrained)
output_dir = model_name
visual_dir = output_dir / "visual"
textual_dir = output_dir / "textual"
to_onnx(config, opset_version, visual_dir, textual_dir, cache=cache)