ll-create / venv /bin /convert-caffe2-to-onnx
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 caffe2_to_onnx
if __name__ == '__main__':
sys.argv[0] = re.sub(r'(-script\.pyw|\.exe)?$', '', sys.argv[0])
sys.exit(caffe2_to_onnx())