Update README.md

README.md

@@ -1,3 +1,24 @@
----
-license: openrail++
----
+---
+license: openrail++
+---
+
+**INFERENTIA 2 ONLY**
+
+```python
+import torch
+from optimum.neuron import NeuronPixArtSigmaPipeline
+
+# Compile
+compiler_args = {"auto_cast": "none"}
+input_shapes = {"batch_size": 1, "height": 512, "width": 512, "sequence_length": 120}
+
+neuron_model = NeuronPixArtSigmaPipeline.from_pretrained("Jingya/pixart_sigma_pipe_xl_2_512_ms", torch_dtype=torch.bfloat16, export=True, disable_neuron_cache=True, **compiler_args, **input_shapes)
+
+# Save locally
+neuron_model.save_pretrained("pixart_sigma_neuron_512/")
+
+# Upload to the HuggingFace Hub
+neuron_model.push_to_hub(
+    "pixart_sigma_neuron_512/", repository_id="optimum/pixart_sigma_pipe_xl_2_512_ms_neuronx"  # Replace with your HF Hub repo id
+)
+```