Update README.md

README.md

@@ -6,8 +6,14 @@ tags:
 
 ```python
 from diffusers import DiffusionPipeline
+pipeline = DiffusionPipeline.from_pretrained("hf-internal-testing/tiny-sdxl-custom-all", trust_remote_code=True)
 
-pipe = DiffusionPipeline.from_pretrained("hf-internal-testing/tiny-stable-diffusion-xl-pipe")
-```
+assert pipeline.config.unet == ('diffusers_modules.local.my_unet_model', 'MyUNetModel')
+assert pipeline.config.scheduler == ('diffusers_modules.local.my_scheduler', 'MyScheduler')
+assert pipeline.__class__.__name__ == "MyPipeline"
 
-The pipeline was created using this [Colab Notebook](https://colab.research.google.com/gist/sayakpaul/a7b986af7e9ea26562eed4ec1410d766/scratchpad.ipynb). 
+pipeline = pipeline.to(torch_device)
+images = pipeline("test", num_inference_steps=2, output_type="np")[0]
+
+assert images.shape == (1, 64, 64, 3)
+```