Update README.md

README.md

@@ -10,7 +10,7 @@ Lumina-Image-2.0 is a 2 billion parameter flow-based diffusion transformer capab
 
 ```python
 import torch
-from diffusers import FluxPipeline
+from diffusers import LuminaText2ImgPipeline
 
 pipe = Lumina2Text2ImgPipeline.from_pretrained("Alpha-VLLM/Lumina-Image-2.0", torch_dtype=torch.bfloat16)
 pipe.enable_model_cpu_offload() #save some VRAM by offloading the model to CPU. Remove this if you have enough GPU power