Fix vit model config.

Signed-off-by: ldwang <ftgreat@gmail.com>
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