ortal1602 commited on
Commit
16fe86c
·
verified ·
1 Parent(s): 2a14d63

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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.th", device=device) # one of ['PAST', 'PAST_streamable']
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
  # ----------------------------------------------------------------------