Spaces:
Runtime error
Runtime error
allow caption to dominate subreddit
Browse files
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])
|