diff options
Diffstat (limited to 'model-integration/src/test/models/onnx/transformer')
-rw-r--r-- | model-integration/src/test/models/onnx/transformer/dummy_transformer.py | 2 | ||||
-rw-r--r-- | model-integration/src/test/models/onnx/transformer/dummy_transformer_without_type_ids.py | 1 |
2 files changed, 2 insertions, 1 deletions
diff --git a/model-integration/src/test/models/onnx/transformer/dummy_transformer.py b/model-integration/src/test/models/onnx/transformer/dummy_transformer.py index 1028035d7c0..2dc91d1468b 100644 --- a/model-integration/src/test/models/onnx/transformer/dummy_transformer.py +++ b/model-integration/src/test/models/onnx/transformer/dummy_transformer.py @@ -1,4 +1,4 @@ -# Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +# Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. import torch import torch.onnx diff --git a/model-integration/src/test/models/onnx/transformer/dummy_transformer_without_type_ids.py b/model-integration/src/test/models/onnx/transformer/dummy_transformer_without_type_ids.py index 4c5f5ebe330..998436bd3e4 100644 --- a/model-integration/src/test/models/onnx/transformer/dummy_transformer_without_type_ids.py +++ b/model-integration/src/test/models/onnx/transformer/dummy_transformer_without_type_ids.py @@ -1,3 +1,4 @@ +# Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. import torch import torch.onnx import torch.nn as nn |