File size: 520 Bytes
5e7b957 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
{
"hmp_bf16_ops": [
"add",
"addmm",
"bmm",
"div",
"dropout",
"gelu",
"iadd",
"linear",
"layer_norm",
"matmul",
"mm",
"rsub",
"softmax",
"truediv"
],
"hmp_fp32_ops": [
"cross_entropy",
"embedding",
"nll_loss",
"log_softmax"
],
"hmp_is_verbose": false,
"hmp_opt_level": "O1",
"optimum_version": "1.4.0",
"transformers_version": "4.23.1",
"use_fused_adam": true,
"use_fused_clip_norm": true,
"use_habana_mixed_precision": true
}
|