Alexandre-Numind commited on
Commit
65ff1ff
1 Parent(s): 4ca9be5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -44,4 +44,5 @@ emb = model(
44
  torch.reshape(torch.tensor(encoding.input_ids),(1,len(encoding.input_ids))).to(device),output_hidden_states=True
45
  ).hidden_states[-1].cpu().detach()
46
 
47
- embText = torch.mean(emb,axis = 1)
 
 
44
  torch.reshape(torch.tensor(encoding.input_ids),(1,len(encoding.input_ids))).to(device),output_hidden_states=True
45
  ).hidden_states[-1].cpu().detach()
46
 
47
+ embText = torch.mean(emb,axis = 1)
48
+ ```