mirror of
https://github.com/ml-explore/mlx.git
synced 2025-09-22 13:28:11 +08:00
Allow quant layer to be unfrozen (#2142)
This commit is contained in:
@@ -193,12 +193,6 @@ class QuantizedLinear(Module):
|
||||
# Freeze this model's parameters
|
||||
self.freeze()
|
||||
|
||||
def unfreeze(self, *args, **kwargs):
|
||||
"""Wrap unfreeze so that we unfreeze any layers we might contain but
|
||||
our parameters will remain frozen."""
|
||||
super().unfreeze(*args, **kwargs)
|
||||
self.freeze(recurse=False)
|
||||
|
||||
def _extra_repr(self):
|
||||
out_dims, in_dims = self.weight.shape
|
||||
in_dims *= 32 // self.bits
|
||||
|
Reference in New Issue
Block a user