Update README.md

README.md

@@ -16,7 +16,7 @@ pip install -U bitsandbytes
 After installing the required library, you can run the following script: 
 
 ```python
-from diffusers import AutoModel, FluxPipeline
+from diffusers import FluxPipeline
 
 pipe = FluxPipeline.from_pretrained(
     "diffusers/FLUX.1-dev-bnb-4bit",
@@ -50,7 +50,7 @@ This checkpoint was created with the following script using "black-forest-labs/F
 ```python
 
 import torch
-from diffusers import AutoModel, FluxPipeline
+from diffusers import FluxPipeline
 from diffusers import BitsAndBytesConfig as DiffusersBitsAndBytesConfig
 from diffusers.quantizers import PipelineQuantizationConfig
 from transformers import BitsAndBytesConfig as TransformersBitsAndBytesConfig