Update README.md

README.md

@@ -24,7 +24,7 @@ This model is recommended to work with **TFT-ID-1.0**[[HF]](https://huggingface.
 # check out https://huggingface.co/microsoft/Phi-3.5-vision-instruct for more details
 
 import torch
-from transformers import AutoModelForCausalLM, AutoProcessor
+from transformers import AutoModelForCausalLM, AutoProcessor, BitsAndBytesConfig
 from PIL import Image
 import requests
 
@@ -38,7 +38,7 @@ model = AutoModelForCausalLM.from_pretrained(
   trust_remote_code=True, 
   torch_dtype="auto", 
   _attn_implementation='flash_attention_2',
-  load_in_4bit=True # Optional: Load model in 4-bit mode to save memory
+  quantization_config=BitsAndBytesConfig(load_in_4bit=True) # Optional: Load model in 4-bit mode to save memory
 )
 
 processor = AutoProcessor.from_pretrained(model_id,