mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-09-03 06:54:33 +08:00
Support for slerp merging models (#455)
* support for slerp merging models * docs * update docs * format'
This commit is contained in:
@@ -216,3 +216,7 @@ class Model(nn.Module):
|
||||
|
||||
y, cache = self.transformer(x, mask, cache)
|
||||
return self.lm_head(y), cache
|
||||
|
||||
@property
|
||||
def layers(self):
|
||||
return self.transformer.h
|
||||
|
Reference in New Issue
Block a user