Issue fix

app.py

@@ -164,6 +164,7 @@ class SkinGPT4(nn.Module):
         self.q_former.eval()
         print("Loaded QFormer")
         self.llama = self._init_llama()
+        self.llama = self.llama.to(device)
         self.llama.resize_token_embeddings(len(self.tokenizer))
 
         self.llama_proj = nn.Linear(