15 lines
335 B
Python
15 lines
335 B
Python
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)
|
|
|
|
|
|
|