bump
This commit is contained in:
14
export_onnx_v2.py
Normal file
14
export_onnx_v2.py
Normal 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)
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user