ivanzhouyq
commited on
Commit
•
88b81c1
1
Parent(s):
51b83bb
update readme and model safetensor
Browse files- README.md +3 -1
- model.safetensors +2 -2
README.md
CHANGED
@@ -19,4 +19,6 @@ torch_out = torch_model(
|
|
19 |
position_ids=None,
|
20 |
)
|
21 |
torch_out = torch.nn.functional.softmax(torch_out.logits, dim=-1)
|
22 |
-
```
|
|
|
|
|
|
19 |
position_ids=None,
|
20 |
)
|
21 |
torch_out = torch.nn.functional.softmax(torch_out.logits, dim=-1)
|
22 |
+
```
|
23 |
+
|
24 |
+
Note that you need to install `safetensors` together with `transformers` and `torch` to make it work.
|
model.safetensors
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:844eb078f8af73181515736354aedcd84d99b1dd21e1218da5e7d4454df46463
|
3 |
+
size 836334888
|