Update README.md
Browse files
README.md
CHANGED
|
@@ -58,7 +58,7 @@ from past.models.past_model import PastModel
|
|
| 58 |
import torch
|
| 59 |
|
| 60 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 61 |
-
model = PastModel.from_pretrained("PAST
|
| 62 |
|
| 63 |
|
| 64 |
# ----------------------------------------------------------------------
|
|
|
|
| 58 |
import torch
|
| 59 |
|
| 60 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 61 |
+
model = PastModel.from_pretrained("PAST", device=device) # one of ['PAST', 'PAST_streamable']
|
| 62 |
|
| 63 |
|
| 64 |
# ----------------------------------------------------------------------
|