Update modeling_cloverlm.py
Browse files- 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 |
|