recursionaut commited on
Commit
5815de0
1 Parent(s): f5d8240

Update config.json (#12)

Browse files

- Update config.json (7dcbbac83de76c1fab808f663af055e0069669af)

Files changed (1) hide show
  1. config.json +4 -0
config.json CHANGED
@@ -1,5 +1,9 @@
1
  {
2
  "_attn_implementation_autoset": true,
 
 
 
 
3
  "apply_loss_unmasked": false,
4
  "architectures": [
5
  "MAEModel"
 
1
  {
2
  "_attn_implementation_autoset": true,
3
+ "auto_map": {
4
+ "AutoModel": "huggingface_mae.MAEModel",
5
+ "AutoConfig": "huggingface_mae.MAEConfig"
6
+ },
7
  "apply_loss_unmasked": false,
8
  "architectures": [
9
  "MAEModel"