add unittest

WAYKEN-TSE · Aug 1, 2023 · 2fad285 · 2fad285
1 parent 0a9cb60
commit 2fad285
Show file tree

Hide file tree

Showing 9 changed files with 2,012 additions and 559 deletions.
diff --git a/paddlevlp/__init__.py b/paddlevlp/__init__.py
@@ -17,3 +17,4 @@
 from .models import *
 from .optimization import *
 from .processors import *
+from .tests import *
diff --git a/paddlevlp/models/blip2/__init__.py b/paddlevlp/models/blip2/__init__.py
@@ -11,3 +11,8 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+
+from .configuration import *
+from .modeling import *
+
+__all__ = configuration.__all__ + modeling.__all__