git clone https://github.com/neuralmagic/sparseml
pip install -e sparseml[clip]
python sparseml/integrations/clip/clip_onnx_export.py --model ViT-B-32 --pretrained laion2b_s34b_b79k --export-path onnx
1import onnx
2from onnx import helper
3from onnx import TensorProto
4
5# Define the new input tensor
6model_path = 'onnx/clip_text.onnx'
7model = onnx.load(model_path)
8new_input = helper.make_tensor_value_info(
9 'input_1',
10 TensorProto.INT64,
11 ['batch']
12)
13model.graph.input.append(new_input)
14onnx.save(model, 'modified_model.onnx')