Update README.md (EasyDeL auto-generated)

README.md

@@ -38,6 +38,8 @@ model = ed.AutoEasyDeLModelForCausalLM.from_pretrained(
     dtype=dtype,
     param_dtype=dtype,
     precision=lax.Precision("fastest"),
+    sharding_axis_names=("dp", "fsdp", "ep", "tp", "sp"),
+    sharding_axis_dims=(1, -1, 1, 1, 1),
     config_kwargs=ed.EasyDeLBaseConfigDict(
         attn_dtype=dtype,
         attn_mechanism=ed.AttentionMechanisms.SPLASH,