Update README.md

README.md

@@ -23,22 +23,22 @@ This pipeline uses the model SDXL and generates images of 1024x1024 resolution.
 
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
 
-``
+``` 
 custom_pipeline = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
     variant="fp16",
     torch_dtype=torch.float16,
     use_safetensors=True,
-    custom_pipeline="syr99/initial_noise_loader",
+    custom_pipeline="syr99/initial_noise_loader"
 ).to(device)
-``
+``` 
 
 The class `NoiseLoaderMixin` can serve for other methods to initialize differently the noise at the first timestep.
 
 ## Citation
 Code adapted from [Github](https://github.com/xie-lab-ml/Golden-Noise-for-Diffusion-Models)
 
-```
+``` 
 @misc{zhou2024goldennoisediffusionmodels,
       title={Golden Noise for Diffusion Models: A Learning Framework}, 
       author={Zikai Zhou and Shitong Shao and Lichen Bai and Zhiqiang Xu and Bo Han and Zeke Xie},
@@ -48,4 +48,4 @@ Code adapted from [Github](https://github.com/xie-lab-ml/Golden-Noise-for-Diffus
       primaryClass={cs.LG},
       url={https://arxiv.org/abs/2411.09502}, 
 }
-```
+```