Upload modeling_mic21.py with huggingface_hub

modeling_mic21.py

@@ -78,7 +78,7 @@ class MIC21SummarizerModel(PreTrainedModel):
             {"role":"system","content":"Generate title and description for the provided image. The image features are: "},
             {"role":"user","content":"Generate a title:"}]
         
-        tokenized_messages = self.components["tokenizer"].apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt")
+        tokenized_messages = self.components["tokenizer"].apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").cuda()
             #.to(self.in_device)     
         vectorized_messages = self.components["llm"].model.embed_tokens(tokenized_messages[0]).unsqueeze(0)
         vectorized_messages = vectorized_messages.repeat(batch_size,1,1)