This commit is contained in:
LYMDLUT 2023-08-25 19:48:23 +02:00 committed by GitHub
commit 3c4f270dff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -144,7 +144,7 @@ class MiniGPT4(Blip2Base):
)
inputs_llama = self.llama_proj(query_output.last_hidden_state)
atts_llama = torch.ones(inputs_llama.size()[:-1], dtype=torch.long).to(image.device)
atts_llama = torch.ones(inputs_llama.size()[:-1], dtype=torch.long).to(device)
return inputs_llama, atts_llama
def prompt_wrap(self, img_embeds, atts_img, prompt):