Update modeling_hf_nomic_bert.py
Browse files
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=
|
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 |
)
|