yagizdevre commited on
Commit
44d4c6b
·
1 Parent(s): ca03f9d

naming fix

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -42,7 +42,7 @@
42
  "buffer": "bfloat16"
43
  },
44
  "fsdp_modules": [
45
- "AttentionLayer"
46
  ],
47
  "use_activation_checkpointing": true,
48
  "use_flash_fft": true,
 
42
  "buffer": "bfloat16"
43
  },
44
  "fsdp_modules": [
45
+ "Attention"
46
  ],
47
  "use_activation_checkpointing": true,
48
  "use_flash_fft": true,