updated post domination

model.py

@@ -83,7 +83,12 @@ class VirTexModel():
             if True: #sub_prompt is not None:
                 cap_tokens = self.tokenizer.encode(prompt)
                 cap_tokens = torch.tensor(cap_tokens, device=self.device).long()
-                subreddit_tokens = torch.cat([subreddit_tokens, cap_tokens])
+                subreddit_tokens = torch.cat(
+                    [
+                        subreddit_tokens,
+                        torch.tensor([self.tokenizer.token_to_id("[SEP]")], device=self.device).long(),
+                        cap_tokens
+                    ])
             
             
         predictions: List[Dict[str, Any]] = []