Fix vit model config.

Signed-off-by: ldwang <[email protected]>
FlagAI-Open · Nov 30, 2023 · 0fc0cc2 · 0fc0cc2
1 parent 1c03300
commit 0fc0cc2
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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