Update README.md

README.md

@@ -31,6 +31,11 @@ def get_dummy_components(num_layers: int = 1, num_single_layers: int = 1):
         pooled_projection_dim=32,
         axes_dims_rope=[4, 4, 8],
     )
+    transformer.save_pretrained(
+        "dummy_flux_sharded_transformer/",
+        safe_serialization=True,
+        max_shard_size="200KB"
+    )
     clip_text_encoder_config = CLIPTextConfig(
         bos_token_id=0,
         eos_token_id=2,
@@ -107,5 +112,4 @@ if __name__ == "__main__":
     components = get_dummy_components()
     pipeline = FluxKontextPipeline(**components)
     pipeline.push_to_hub("hf-internal-testing/tiny-flux-kontext-pipe-gated-silu")
-
 ```