fix

example_code.py

@@ -22,7 +22,7 @@ def load_model_from_huggingface(repo_id):
     model_path = Path(model_path)
 
     tokenizer, model, image_processor, context_len = load_pretrained_model(model_path, model_base='liuhaotian/llava-v1.5-7b',
-                                                                           model_name="llava-v1.5-7b-task-lora_radialog_instruct_llava_biovil_unfrozen_2e-5_5epochs_v5_checkpoint-21000", load_8bit=False, $
+                                                                           model_name="llava-v1.5-7b-task-lora_radialog_instruct_llava_biovil_unfrozen_2e-5_5epochs_v5_checkpoint-21000", load_8bit=False, load_4bit=False)
 
 
     return tokenizer, model, image_processor, context_len
@@ -51,7 +51,7 @@ if __name__ == '__main__':
     findings = ', '.join(findings).lower().strip()
 
     conv = conv_vicuna_v1.copy()
-    REPORT_GEN_PROMPT = f"<image>. Predicted Findings: {findings}. You are to act as a radiologist and write the finding section of a chest x-ray radiology report for this X-ray image and the given predi$
+    REPORT_GEN_PROMPT = f"<image>. Predicted Findings: {findings}. You are to act as a radiologist and write the finding section of a chest x-ray radiology report for this X-ray image and the given predicted findings. Write in the style of a radiologist, write one fluent text without enumeration, be concise and don't provide explanations or reasons."
     print("USER: ", REPORT_GEN_PROMPT)
     conv.append_message("USER", REPORT_GEN_PROMPT)
     conv.append_message("ASSISTANT", None)