Alexandre-Numind
commited on
Commit
•
65ff1ff
1
Parent(s):
4ca9be5
Update README.md
Browse files
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 |
+
```
|