mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-12-16 02:08:55 +08:00
reformatted
This commit is contained in:
@@ -91,7 +91,7 @@ class FeedForward(nn.Module):
|
||||
class MOEFeedForward(nn.Module):
|
||||
def __init__(self, args: ModelArgs):
|
||||
super().__init__()
|
||||
|
||||
|
||||
if args.moe is None:
|
||||
raise ValueError("args.moe must not be None for MOEFeedForward")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user