reformatted

This commit is contained in:
paramthakkar123
2025-04-04 07:36:47 +05:30
parent d7cab9d5f5
commit 298178d669
2 changed files with 3 additions and 2 deletions

View File

@@ -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")