Fill-Mask
Transformers
PyTorch
Safetensors
English
nomic_bert
custom_code
zpn commited on
Commit
e5042dc
·
verified ·
1 Parent(s): 359596a

Update modeling_hf_nomic_bert.py

Browse files
Files changed (1) hide show
  1. modeling_hf_nomic_bert.py +1 -1
modeling_hf_nomic_bert.py CHANGED
@@ -1622,7 +1622,7 @@ class NomicBertBlock(NomicBertPreTrainedModel):
1622
  moe_normalize_expert_weights=config.moe_normalize_expert_weights,
1623
  activation_fn=activation,
1624
  mlp_type="glu" if config.activation_function == "swiglu" else "mlp",
1625
- fp16=True,
1626
  bf16=False,
1627
  return_bias=False,
1628
  )
 
1622
  moe_normalize_expert_weights=config.moe_normalize_expert_weights,
1623
  activation_fn=activation,
1624
  mlp_type="glu" if config.activation_function == "swiglu" else "mlp",
1625
+ fp16=False,
1626
  bf16=False,
1627
  return_bias=False,
1628
  )