mansaripo commited on
Commit
d11440f
·
verified ·
1 Parent(s): fb4181d

Update modeling_cloverlm.py

Browse files
Files changed (1) hide show
  1. modeling_cloverlm.py +2 -0
modeling_cloverlm.py CHANGED
@@ -32,6 +32,8 @@ def _make_linear(in_f, out_f, bias, quartet_2_impl):
32
  raise e
33
 
34
  return Quartet_II_linear(in_f, out_f, bias)
 
 
35
  else:
36
  raise ValueError(f"Unsupported quartet_2_impl: {quartet_2_impl}")
37
 
 
32
  raise e
33
 
34
  return Quartet_II_linear(in_f, out_f, bias)
35
+ elif quartet_2_impl in ("bf16", None, ""):
36
+ return nn.Linear(in_f, out_f, bias=bias)
37
  else:
38
  raise ValueError(f"Unsupported quartet_2_impl: {quartet_2_impl}")
39