Update README.md

README.md

@@ -46,7 +46,7 @@ import torch
 from diffusers import FluxPipeline
 
 pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
-pipe.enable_model_cpu_offload()
+pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power
 
 prompt = "A cat holding a sign that says hello world"
 image = pipe(
@@ -60,6 +60,8 @@ image = pipe(
 image.save("flux-schnell.png")
 ```
 
+To learn mode check out the [diffusers](https://huggingface.co/docs/diffusers/main/en/api/pipelines/flux) documentation
+
 ---
 # Limitations
 - This model is not intended or able to provide factual information.