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

allow caption to dominate subreddit

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 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])