Skip to content

Commit

Permalink
change opset_version to 12
Browse files Browse the repository at this point in the history
  • Loading branch information
Lednik7 committed May 3, 2022
1 parent 49e63d7 commit fd61389
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion clip_onnx/utils.py
Expand Up @@ -49,6 +49,6 @@ def attention(self, x: torch.Tensor):


DEFAULT_EXPORT = dict(input_names=['input'], output_names=['output'],
export_params=True, verbose=False, opset_version=14,
export_params=True, verbose=False, opset_version=12,
do_constant_folding=True,
dynamic_axes={'input': {0: 'batch_size'}, 'output': {0: 'batch_size'}})

0 comments on commit fd61389

Please sign in to comment.