BUAADreamer hiyouga commited on
Commit
52df825
1 Parent(s): 6d6b652

Update config.json (#1)

Browse files

- Update config.json (fa0185f8fab5714f5cfc9142b6755cd7706e20fb)


Co-authored-by: Yaowei Zheng <hiyouga@users.noreply.huggingface.co>

Files changed (1) hide show
  1. config.json +2 -3
config.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "_name_or_path": "saves/paligemma-3b-chat-v0.2",
3
  "architectures": [
4
  "PaliGemmaForConditionalGeneration"
5
  ],
@@ -19,7 +18,7 @@
19
  "num_hidden_layers": 18,
20
  "num_image_tokens": 1024,
21
  "num_key_value_heads": 1,
22
- "torch_dtype": "float32",
23
  "vocab_size": 257216
24
  },
25
  "torch_dtype": "bfloat16",
@@ -39,4 +38,4 @@
39
  "vision_use_head": false
40
  },
41
  "vocab_size": 257216
42
- }
 
1
  {
 
2
  "architectures": [
3
  "PaliGemmaForConditionalGeneration"
4
  ],
 
18
  "num_hidden_layers": 18,
19
  "num_image_tokens": 1024,
20
  "num_key_value_heads": 1,
21
+ "torch_dtype": "bfloat16",
22
  "vocab_size": 257216
23
  },
24
  "torch_dtype": "bfloat16",
 
38
  "vision_use_head": false
39
  },
40
  "vocab_size": 257216
41
+ }