Update modeling_quiet.py
Browse files- modeling_quiet.py +1 -0
modeling_quiet.py
CHANGED
|
@@ -1312,6 +1312,7 @@ class QuietForCausalLM(QuietPreTrainedModel, GenerationMixin):
|
|
| 1312 |
nn.init.xavier_uniform_(module.weight)
|
| 1313 |
|
| 1314 |
@torch.no_grad()
|
|
|
|
| 1315 |
def infer(
|
| 1316 |
self,
|
| 1317 |
input_ids: torch.LongTensor,
|
|
|
|
| 1312 |
nn.init.xavier_uniform_(module.weight)
|
| 1313 |
|
| 1314 |
@torch.no_grad()
|
| 1315 |
+
|
| 1316 |
def infer(
|
| 1317 |
self,
|
| 1318 |
input_ids: torch.LongTensor,
|