fix

allenai · Nov 22, 2024 · f32b35b · f32b35b
1 parent cc3308a
commit f32b35b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/test/nn/transformer/model_test.py b/src/test/nn/transformer/model_test.py
@@ -101,7 +101,7 @@ def run_ngpt_with_fsdp2():
         dp_config=TransformerDataParallelConfig(name=DataParallelType.fsdp),
     )
     model = config.build(init_device="meta", max_seq_len=1024)
-    optim = torch.optim.Adam(model.params())
+    optim = torch.optim.Adam(model.parameters())
 
     # Take an optimizer step.
     model(input_ids=torch.randint(0, 50257, (2, 128))).sum().backward()