farbodpya commited on
Commit
7207941
·
verified ·
1 Parent(s): d715af6

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -1
config.json CHANGED
@@ -6,5 +6,7 @@
6
  "nhead": 16,
7
  "num_layers": 8,
8
  "dropout": 0.2,
9
- "adaptive_height": 8
 
 
10
  }
 
6
  "nhead": 16,
7
  "num_layers": 8,
8
  "dropout": 0.2,
9
+ "adaptive_height": 8,
10
+ "torch_dtype": "float32",
11
+ "transformers_version": "4.44.0"
12
  }