Update README.md

README.md

@@ -28,6 +28,9 @@ pip install torch transformers peft
 ```
 ## Usage (Transformers)
 ```python
+import torch
+import torch.nn.functional as F
+from transformers import AutoModel, AutoTokenizer
 model_id = "SkywalkerLu/TransHLA2.0-IM"
 model = AutoModel.from_pretrained(model_id, trust_remote_code=True).to(device).eval()
 ```