From ca72f5329c921a179c7c7b624e497adbee66c4bc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=A1=8C=E5=97=94?= Date: Tue, 18 Oct 2022 16:39:52 +0800 Subject: [PATCH] commit code --- modelscope/metrics/accuracy_metric.py | 2 ++ modelscope/models/multi_modal/ofa/adaptor/__init__.py | 0 modelscope/models/multi_modal/ofa/modeling_ofa.py | 1 + modelscope/trainers/multi_modal/ofa/__init__.py | 2 ++ modelscope/trainers/multi_modal/ofa/ofa_trainer.py | 2 ++ 5 files changed, 7 insertions(+) create mode 100644 modelscope/models/multi_modal/ofa/adaptor/__init__.py diff --git a/modelscope/metrics/accuracy_metric.py b/modelscope/metrics/accuracy_metric.py index aab9a138..1761786e 100644 --- a/modelscope/metrics/accuracy_metric.py +++ b/modelscope/metrics/accuracy_metric.py @@ -1,3 +1,5 @@ +# Copyright (c) Alibaba, Inc. and its affiliates. + from typing import Dict import numpy as np diff --git a/modelscope/models/multi_modal/ofa/adaptor/__init__.py b/modelscope/models/multi_modal/ofa/adaptor/__init__.py new file mode 100644 index 00000000..e69de29b diff --git a/modelscope/models/multi_modal/ofa/modeling_ofa.py b/modelscope/models/multi_modal/ofa/modeling_ofa.py index bc749b46..0a7a2ce6 100755 --- a/modelscope/models/multi_modal/ofa/modeling_ofa.py +++ b/modelscope/models/multi_modal/ofa/modeling_ofa.py @@ -54,6 +54,7 @@ OFA_PRETRAINED_MODEL_ARCHIVE_LIST = [ 'ofa-medium', 'ofa-base', 'ofa-large', + 'ofa-huge', ] try: diff --git a/modelscope/trainers/multi_modal/ofa/__init__.py b/modelscope/trainers/multi_modal/ofa/__init__.py index 7222c48c..34e4ec7a 100644 --- a/modelscope/trainers/multi_modal/ofa/__init__.py +++ b/modelscope/trainers/multi_modal/ofa/__init__.py @@ -1 +1,3 @@ +# Copyright (c) Alibaba, Inc. and its affiliates. + from .ofa_trainer import OFATrainer diff --git a/modelscope/trainers/multi_modal/ofa/ofa_trainer.py b/modelscope/trainers/multi_modal/ofa/ofa_trainer.py index 5c65a129..3daadf43 100644 --- a/modelscope/trainers/multi_modal/ofa/ofa_trainer.py +++ b/modelscope/trainers/multi_modal/ofa/ofa_trainer.py @@ -1,3 +1,5 @@ +# Copyright (c) Alibaba, Inc. and its affiliates. + import math import os from functools import partial