Update README.md
Browse files
README.md
CHANGED
|
@@ -3,6 +3,7 @@ library_name: peft
|
|
| 3 |
license: mit
|
| 4 |
datasets:
|
| 5 |
- multi_nli
|
|
|
|
| 6 |
language:
|
| 7 |
- en
|
| 8 |
metrics:
|
|
@@ -75,5 +76,4 @@ for k, v in tok.items():
|
|
| 75 |
tok[k] = v.cuda()
|
| 76 |
vec = model(output_hidden_states=True, **tok).hidden_states[-1][:, -1].float().detach().cpu().numpy()
|
| 77 |
print(vec)
|
| 78 |
-
```
|
| 79 |
-
|
|
|
|
| 3 |
license: mit
|
| 4 |
datasets:
|
| 5 |
- multi_nli
|
| 6 |
+
- snli
|
| 7 |
language:
|
| 8 |
- en
|
| 9 |
metrics:
|
|
|
|
| 76 |
tok[k] = v.cuda()
|
| 77 |
vec = model(output_hidden_states=True, **tok).hidden_states[-1][:, -1].float().detach().cpu().numpy()
|
| 78 |
print(vec)
|
| 79 |
+
```
|
|
|