Update README.md

README.md

@@ -45,6 +45,7 @@ This dataset enables Bee-8B to achieve exceptional performance, particularly in
 
 ```python
 import requests
+import torch
 from PIL import Image
 from transformers import AutoModel, AutoProcessor
 
@@ -53,7 +54,7 @@ model_path = "Open-Bee/Bee-8B-RL"
 # Load model
 model = AutoModel.from_pretrained(
     model_path,
-    dtype="bfloat16",
+    torch_dtype=torch.bfloat16,
     trust_remote_code=True,
 ).to("cuda")