miguelcarv commited on
Commit
9d69583
1 Parent(s): 71cc30c

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +3 -3
config.json CHANGED
@@ -7,7 +7,7 @@
7
  "cross_interval": 2,
8
  "from_layer": 0,
9
  "reduce": 1,
10
- "encoder_dtype": "torch.bfloat16",
11
- "decoder_dtype": "torch.bfloat16",
12
- "other_params_dtype": "torch.bfloat16"
13
  }
 
7
  "cross_interval": 2,
8
  "from_layer": 0,
9
  "reduce": 1,
10
+ "encoder_dtype": "torch.float32",
11
+ "decoder_dtype": "torch.float32",
12
+ "other_params_dtype": "torch.float32"
13
  }