fix init

fix muon to recieve model parameter, not model

torch-ext/optimizer/muon.py

@@ -210,7 +210,7 @@ class Muon(torch.optim.Optimizer):
 
     def __init__(
         self,
-        model,
+        params,
         is_muon_func=default_is_muon,
         lr=1e-3,
         momentum=0.95,
@@ -233,7 +233,7 @@ class Muon(torch.optim.Optimizer):
             none_grad=none_grad,
         )
 
-        super().__init__(model.parameters(), defaults)
+        super().__init__(params, defaults)
         self.is_muon_func = is_muon_func
         self.model = model