Spaces:
Runtime error
Runtime error
who knew you needed : for if statments, not me
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 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])
|