From 0fc0cc2fb70cbc151119929cb440922802b241d5 Mon Sep 17 00:00:00 2001
From: ldwang <ftgreat@gmail.com>
Date: Thu, 30 Nov 2023 08:45:39 +0800
Subject: [PATCH] Fix vit model config.

Signed-off-by: ldwang <ftgreat@gmail.com>
---
 flagai/model/vision/vit.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/flagai/model/vision/vit.py b/flagai/model/vision/vit.py
index 59465083..ebe6c400 100755
--- a/flagai/model/vision/vit.py
+++ b/flagai/model/vision/vit.py
@@ -228,6 +228,7 @@ def __init__(
         super().__init__(config)
         embed_layer=PatchEmbed
         block_fn=Block
+        config = config.json_config
         vit_config = VitConfig(**config)
         vit_config.num_classes = num_classes
         # config = vit_config