zamborg commited on
Commit
250dc27
1 Parent(s): 16484d3

i dummy pt3 -> forgot to convert to tensor

Browse files
Files changed (1) hide show
  1. model.py +1 -0
model.py CHANGED
@@ -60,6 +60,7 @@ class VirTexModel():
60
  self.tokenizer.encode(subreddit_tokens) +
61
  [self.tokenizer.token_to_id("[SEP]")]
62
  )
 
63
 
64
 
65
  predictions: List[Dict[str, Any]] = []
 
60
  self.tokenizer.encode(subreddit_tokens) +
61
  [self.tokenizer.token_to_id("[SEP]")]
62
  )
63
+ subreddit_tokens = torch.tensor(subreddit_tokens, device=self.device).long()
64
 
65
 
66
  predictions: List[Dict[str, Any]] = []