Remove unnecessary write_tensors()

This commit is contained in:
Yoshi Suhara 2024-08-08 09:38:27 -07:00
parent 45e9d164ac
commit 147cdf641a

View file

@ -3777,16 +3777,6 @@ class NemotronModel(Model):
return [(self.map_tensor_name(name), data_torch)] return [(self.map_tensor_name(name), data_torch)]
def write_tensors(self):
super().write_tensors()
if self._experts is not None:
# flatten `list[dict[str, Tensor]]` into `list[str]`
experts = [k for d in self._experts for k in d.keys()]
if len(experts) > 0:
raise ValueError(f"Unprocessed experts: {experts}")
###### CONVERSION LOGIC ###### ###### CONVERSION LOGIC ######