Update README.md

README.md

@@ -57,6 +57,7 @@ tokenizer = AutoTokenizer.from_pretrained('IDEA-CCNL/Erlangshen-SimCSE-110M-Chin
 ### 使用示例 Usage Examples
 
 ```python
+import torch
 from sklearn.metrics.pairwise import cosine_similarity
 
 texta = '今天天气真不错，我们去散步吧！'
@@ -71,8 +72,10 @@ outputs_b = model(**inputs_b ,output_hidden_states=True)
 textb_embedding = outputs_b.hidden_states[-1][:,0,:].squeeze()
 
 # if you use cuda, the text_embedding should be textb_embedding.cpu().numpy()
-silimarity_soce = cosine_similarity(texta_embedding.reshape(1,-1),textb_embedding .reshape(1,-1))[0][0]
-
+# 或者用torch.no_grad():
+with torch.no_grad():
+    silimarity_soce = cosine_similarity(texta_embedding.reshape(1,-1),textb_embedding .reshape(1,-1))[0][0]
+print(silimarity_soce)
 ```
 
 ## 引用 Citation