ll-create / venv /bin /convert-onnx-to-caffe2
asgeorges's picture
Upload folder using huggingface_hub
2fdce3c
raw
history blame contribute delete
271 Bytes
#!/workspace/kohya_ss/venv/bin/python
# -*- coding: utf-8 -*-
import re
import sys
from caffe2.python.onnx.bin.conversion import onnx_to_caffe2
if __name__ == '__main__':
sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
sys.exit(onnx_to_caffe2())