Fix KeyError raised in Module.update.

e.g., load safetensors from black-forest-labs/FLUX.1-Depth-dev-lora.

python/mlx/nn/layers/base.py

@@ -322,6 +322,10 @@ def apply(dst, parameters):
                         elif isinstance(current_value, (dict, list)):
                             apply(current_value, new_value)
             elif isinstance(parameters, list):
+                if isinstance(dst, dict):
+                    from mlx.nn.layers.containers import Sequential
+                    if isinstance(dst, Sequential):
+                        dst = dst["layers"]
                 for i in range(len(parameters)):
                     current_value = dst[i]
                     new_value = parameters[i]