Update modeling_quiet.py

modeling_quiet.py

@@ -1311,12 +1311,12 @@ class QuietForCausalLM(QuietPreTrainedModel):
         return self.model
     
     def _init_weights(self, module):
-    if isinstance(module, nn.Linear):
-        nn.init.xavier_uniform_(module.weight)
-        if module.bias is not None:
-            nn.init.constant_(module.bias, 0)
-    elif isinstance(module, nn.Embedding):
-        nn.init.xavier_uniform_(module.weight)
+        if isinstance(module, nn.Linear):
+            nn.init.xavier_uniform_(module.weight)
+            if module.bias is not None:
+                nn.init.constant_(module.bias, 0)
+        elif isinstance(module, nn.Embedding):
+            nn.init.xavier_uniform_(module.weight)
 
     @torch.no_grad()
     def infer(