fix added

SkinGPT.py

@@ -253,10 +253,15 @@ class SkinGPT4(nn.Module):
         # ### Response:
         # """
 
-        prompt = """### Skin Diagnosis Analysis ###
+        # prompt = """### Skin Diagnosis Analysis ###
+        # <IMAGE>
+        # Could you describe the skin condition in this image?
+        # Please provide a detailed analysis including possible diagnoses.
+        # ### Response:"""
+
+        prompt = """### Instruction:
         <IMAGE>
         Could you describe the skin condition in this image?
-        Please provide a detailed analysis including possible diagnoses.
         ### Response:"""
 
 
@@ -314,6 +319,18 @@ class SkinGPT4(nn.Module):
         print(f"\n[DEBUG] After replacement:")
         print(f"Image token embedding (after):\n{input_embeddings[0, replace_positions[0][1], :5]}...")
 
+        # outputs = self.llama.generate(
+        #     inputs_embeds=input_embeddings,
+        #     max_new_tokens=max_new_tokens,
+        #     temperature=0.7,
+        #     top_k=40,
+        #     top_p=0.9,
+        #     repetition_penalty=1.1,
+        #     do_sample=True,
+        #     pad_token_id = self.tokenizer.eos_token_id,
+        #     eos_token_id = self.tokenizer.eos_token_id
+        # )
+
         outputs = self.llama.generate(
             inputs_embeds=input_embeddings,
             max_new_tokens=max_new_tokens,
@@ -322,8 +339,6 @@ class SkinGPT4(nn.Module):
             top_p=0.9,
             repetition_penalty=1.1,
             do_sample=True,
-            pad_token_id = self.tokenizer.eos_token_id,
-            eos_token_id = self.tokenizer.eos_token_id
         )