zamborg commited on
Commit
99ca89b
1 Parent(s): 76b5b01

who knew you needed : for if statments, not me

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -65,7 +65,7 @@ class VirTexModel():
65
  if prompt is not "":
66
  # at present prompts without subreddits will break without this change
67
  # TODO FIX
68
- if True #sub_prompt is not None:
69
  cap_tokens = self.tokenizer.encode(prompt)
70
  cap_tokens = torch.tensor(cap_tokens, device=self.device).long()
71
  subreddit_tokens = torch.cat([subreddit_tokens, cap_tokens])
 
65
  if prompt is not "":
66
  # at present prompts without subreddits will break without this change
67
  # TODO FIX
68
+ if True: #sub_prompt is not None:
69
  cap_tokens = self.tokenizer.encode(prompt)
70
  cap_tokens = torch.tensor(cap_tokens, device=self.device).long()
71
  subreddit_tokens = torch.cat([subreddit_tokens, cap_tokens])