Update modeling_baichuan.py (#6)

- Update modeling_baichuan.py (0498e281c48996cb9ff7b3f1485830cdb626ef58)


Co-authored-by: JaheimLee <[email protected]>

modeling_baichuan.py

@@ -513,7 +513,7 @@ class NormHead(nn.Module):
             norm_weight = nn.functional.normalize(self.weight)
         elif self.first_flag:
             self.first_flag = False
-            self.weight = nn.Parameter(nn.functional.normalize(self.weight))
+            self.weight.data = nn.functional.normalize(self.weight)
             norm_weight = self.weight
         else:
             norm_weight = self.weight