├── .github └── ISSUE_TEMPLATE │ ├── 1-usage.yaml │ ├── 2-feature-request.yaml │ ├── 3-question.yaml │ └── 4-discussion.yaml ├── .gitignore ├── LICENSE ├── MODEL_CARD.md ├── README.md ├── docs ├── Customize_Component.md ├── Data.md ├── LLaVA_Bench.md ├── LLaVA_from_LLaMA2.md ├── LoRA.md ├── MODEL_ZOO.md └── Release_Notes.md ├── finetune.sh ├── images ├── babydoctor_cute.png ├── chat.png ├── io.png ├── llava_example_cmp.png └── llava_logo.png ├── llava ├── __init__.py ├── constants.py ├── conversation.py ├── eval │ ├── eval_gpt_review.py │ ├── eval_gpt_review_bench.py │ ├── eval_gpt_review_visual.py │ ├── eval_science_qa.py │ ├── eval_science_qa_gpt4.py │ ├── eval_science_qa_gpt4_requery.py │ ├── generate_webpage_data_from_table.py │ ├── model_qa.py │ ├── model_vqa.py │ ├── model_vqa_science.py │ ├── qa_baseline_gpt35.py │ ├── run_llava.py │ ├── summarize_gpt_review.py │ ├── table │ │ ├── answer │ │ │ ├── answer_alpaca-13b.jsonl │ │ │ ├── answer_bard.jsonl │ │ │ ├── answer_gpt35.jsonl │ │ │ ├── answer_llama-13b.jsonl │ │ │ └── answer_vicuna-13b.jsonl │ │ ├── caps_boxes_coco2014_val_80.jsonl │ │ ├── model.jsonl │ │ ├── prompt.jsonl │ │ ├── question.jsonl │ │ ├── results │ │ │ └── test_sqa_llava_13b_v0.json │ │ ├── review │ │ │ ├── review_alpaca-13b_vicuna-13b.jsonl │ │ │ ├── review_bard_vicuna-13b.jsonl │ │ │ ├── review_gpt35_vicuna-13b.jsonl │ │ │ └── review_llama-13b_vicuna-13b.jsonl │ │ ├── reviewer.jsonl │ │ └── rule.json │ └── webpage │ │ ├── figures │ │ ├── alpaca.png │ │ ├── bard.jpg │ │ ├── chatgpt.svg │ │ ├── llama.jpg │ │ ├── swords_FILL0_wght300_GRAD0_opsz48.svg │ │ └── vicuna.jpeg │ │ ├── index.html │ │ ├── script.js │ │ └── styles.css ├── mm_utils.py ├── model │ ├── __init__.py │ ├── apply_delta.py │ ├── builder.py │ ├── consolidate.py │ ├── language_model │ │ ├── llava_llama.py │ │ ├── llava_mpt.py │ │ └── mpt │ │ │ ├── adapt_tokenizer.py │ │ │ ├── attention.py │ │ │ ├── blocks.py │ │ │ ├── configuration_mpt.py │ │ │ ├── custom_embedding.py │ │ │ ├── flash_attn_triton.py │ │ │ ├── hf_prefixlm_converter.py │ │ │ ├── meta_init_context.py │ │ │ ├── modeling_mpt.py │ │ │ ├── norm.py │ │ │ └── param_init_fns.py │ ├── llava_arch.py │ ├── make_delta.py │ ├── merger.py │ ├── multimodal_encoder │ │ ├── builder.py │ │ └── clip_encoder.py │ └── utils.py ├── serve │ ├── __init__.py │ ├── cli.py │ ├── controller.py │ ├── examples │ │ ├── ROCO_00027.jpg │ │ ├── ROCO_00278.jpg │ │ ├── ROCO_01259.jpg │ │ ├── ROCO_01747.jpg │ │ └── ROCO_03257.jpg │ ├── gradio_web_server.py │ ├── model_worker.py │ ├── register_worker.py │ └── test_message.py ├── train │ ├── llama_flash_attn_monkey_patch.py │ ├── llava_trainer.py │ ├── train.py │ └── train_mem.py └── utils.py ├── llmenv.yml ├── playground └── data │ ├── coco2014_val_gpt4_qa_30x3.jsonl │ ├── coco2014_val_qa_eval │ ├── qa90_gpt4_answer.jsonl │ └── qa90_questions.jsonl │ └── prompts │ ├── complex_reasoning │ ├── 000_caps.txt │ ├── 000_conv.txt │ ├── 001_caps.txt │ ├── 001_conv.txt │ ├── 002_caps.txt │ ├── 002_conv.txt │ └── system_message.txt │ ├── conversation │ ├── 000_caps.txt │ ├── 000_conv.txt │ ├── 001_caps.txt │ ├── 001_conv.txt │ └── system_message.txt │ └── detail_description │ ├── 000_caps.txt │ ├── 000_conv.txt │ ├── 001_caps.txt │ ├── 001_conv.txt │ ├── 002_caps.txt │ ├── 002_conv.txt │ └── system_message.txt ├── pyproject.toml └── scripts ├── convert_sqa_to_llava.py ├── convert_sqa_to_llava_base_prompt.py ├── copy_model.py ├── extract_mm_projector.py ├── filter_missing.py ├── finetune.sh ├── finetune_lora.sh ├── finetune_qlora.sh ├── inference.sh ├── load_from_hf.py ├── massage_data.py ├── merge.sh ├── merge_lora_weights.py ├── misc └── inspect.py ├── pretrain.sh ├── quantise.ipynb ├── quantise.py ├── sqa_eval_batch.sh ├── sqa_eval_gather.sh ├── unlink.sh ├── upload_dataset.py ├── upload_lora.py ├── zero2.json ├── zero3.json └── zero3_offload.json /.github/ISSUE_TEMPLATE/1-usage.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/.github/ISSUE_TEMPLATE/1-usage.yaml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/2-feature-request.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/.github/ISSUE_TEMPLATE/2-feature-request.yaml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/3-question.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/.github/ISSUE_TEMPLATE/3-question.yaml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/4-discussion.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/.github/ISSUE_TEMPLATE/4-discussion.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_CARD.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/MODEL_CARD.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/README.md -------------------------------------------------------------------------------- /docs/Customize_Component.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/docs/Customize_Component.md -------------------------------------------------------------------------------- /docs/Data.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/docs/Data.md -------------------------------------------------------------------------------- /docs/LLaVA_Bench.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/docs/LLaVA_Bench.md -------------------------------------------------------------------------------- /docs/LLaVA_from_LLaMA2.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/docs/LLaVA_from_LLaMA2.md -------------------------------------------------------------------------------- /docs/LoRA.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/docs/LoRA.md -------------------------------------------------------------------------------- /docs/MODEL_ZOO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/docs/MODEL_ZOO.md -------------------------------------------------------------------------------- /docs/Release_Notes.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/docs/Release_Notes.md -------------------------------------------------------------------------------- /finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/finetune.sh -------------------------------------------------------------------------------- /images/babydoctor_cute.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/images/babydoctor_cute.png -------------------------------------------------------------------------------- /images/chat.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/images/chat.png -------------------------------------------------------------------------------- /images/io.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/images/io.png -------------------------------------------------------------------------------- /images/llava_example_cmp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/images/llava_example_cmp.png -------------------------------------------------------------------------------- /images/llava_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/images/llava_logo.png -------------------------------------------------------------------------------- /llava/__init__.py: -------------------------------------------------------------------------------- 1 | from .model import LlavaLlamaForCausalLM 2 | -------------------------------------------------------------------------------- /llava/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/constants.py -------------------------------------------------------------------------------- /llava/conversation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/conversation.py -------------------------------------------------------------------------------- /llava/eval/eval_gpt_review.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/eval_gpt_review.py -------------------------------------------------------------------------------- /llava/eval/eval_gpt_review_bench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/eval_gpt_review_bench.py -------------------------------------------------------------------------------- /llava/eval/eval_gpt_review_visual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/eval_gpt_review_visual.py -------------------------------------------------------------------------------- /llava/eval/eval_science_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/eval_science_qa.py -------------------------------------------------------------------------------- /llava/eval/eval_science_qa_gpt4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/eval_science_qa_gpt4.py -------------------------------------------------------------------------------- /llava/eval/eval_science_qa_gpt4_requery.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/eval_science_qa_gpt4_requery.py -------------------------------------------------------------------------------- /llava/eval/generate_webpage_data_from_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/generate_webpage_data_from_table.py -------------------------------------------------------------------------------- /llava/eval/model_qa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/model_qa.py -------------------------------------------------------------------------------- /llava/eval/model_vqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/model_vqa.py -------------------------------------------------------------------------------- /llava/eval/model_vqa_science.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/model_vqa_science.py -------------------------------------------------------------------------------- /llava/eval/qa_baseline_gpt35.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/qa_baseline_gpt35.py -------------------------------------------------------------------------------- /llava/eval/run_llava.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/run_llava.py -------------------------------------------------------------------------------- /llava/eval/summarize_gpt_review.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/summarize_gpt_review.py -------------------------------------------------------------------------------- /llava/eval/table/answer/answer_alpaca-13b.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/answer/answer_alpaca-13b.jsonl -------------------------------------------------------------------------------- /llava/eval/table/answer/answer_bard.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/answer/answer_bard.jsonl -------------------------------------------------------------------------------- /llava/eval/table/answer/answer_gpt35.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/answer/answer_gpt35.jsonl -------------------------------------------------------------------------------- /llava/eval/table/answer/answer_llama-13b.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/answer/answer_llama-13b.jsonl -------------------------------------------------------------------------------- /llava/eval/table/answer/answer_vicuna-13b.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/answer/answer_vicuna-13b.jsonl -------------------------------------------------------------------------------- /llava/eval/table/caps_boxes_coco2014_val_80.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/caps_boxes_coco2014_val_80.jsonl -------------------------------------------------------------------------------- /llava/eval/table/model.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/model.jsonl -------------------------------------------------------------------------------- /llava/eval/table/prompt.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/prompt.jsonl -------------------------------------------------------------------------------- /llava/eval/table/question.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/question.jsonl -------------------------------------------------------------------------------- /llava/eval/table/results/test_sqa_llava_13b_v0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/results/test_sqa_llava_13b_v0.json -------------------------------------------------------------------------------- /llava/eval/table/review/review_alpaca-13b_vicuna-13b.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/review/review_alpaca-13b_vicuna-13b.jsonl -------------------------------------------------------------------------------- /llava/eval/table/review/review_bard_vicuna-13b.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/review/review_bard_vicuna-13b.jsonl -------------------------------------------------------------------------------- /llava/eval/table/review/review_gpt35_vicuna-13b.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/review/review_gpt35_vicuna-13b.jsonl -------------------------------------------------------------------------------- /llava/eval/table/review/review_llama-13b_vicuna-13b.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/review/review_llama-13b_vicuna-13b.jsonl -------------------------------------------------------------------------------- /llava/eval/table/reviewer.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/reviewer.jsonl -------------------------------------------------------------------------------- /llava/eval/table/rule.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/table/rule.json -------------------------------------------------------------------------------- /llava/eval/webpage/figures/alpaca.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/webpage/figures/alpaca.png -------------------------------------------------------------------------------- /llava/eval/webpage/figures/bard.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/webpage/figures/bard.jpg -------------------------------------------------------------------------------- /llava/eval/webpage/figures/chatgpt.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/webpage/figures/chatgpt.svg -------------------------------------------------------------------------------- /llava/eval/webpage/figures/llama.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/webpage/figures/llama.jpg -------------------------------------------------------------------------------- /llava/eval/webpage/figures/swords_FILL0_wght300_GRAD0_opsz48.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/webpage/figures/swords_FILL0_wght300_GRAD0_opsz48.svg -------------------------------------------------------------------------------- /llava/eval/webpage/figures/vicuna.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/webpage/figures/vicuna.jpeg -------------------------------------------------------------------------------- /llava/eval/webpage/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/webpage/index.html -------------------------------------------------------------------------------- /llava/eval/webpage/script.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/webpage/script.js -------------------------------------------------------------------------------- /llava/eval/webpage/styles.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/eval/webpage/styles.css -------------------------------------------------------------------------------- /llava/mm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/mm_utils.py -------------------------------------------------------------------------------- /llava/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/__init__.py -------------------------------------------------------------------------------- /llava/model/apply_delta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/apply_delta.py -------------------------------------------------------------------------------- /llava/model/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/builder.py -------------------------------------------------------------------------------- /llava/model/consolidate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/consolidate.py -------------------------------------------------------------------------------- /llava/model/language_model/llava_llama.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/llava_llama.py -------------------------------------------------------------------------------- /llava/model/language_model/llava_mpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/llava_mpt.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/adapt_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/adapt_tokenizer.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/attention.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/blocks.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/configuration_mpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/configuration_mpt.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/custom_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/custom_embedding.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/flash_attn_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/flash_attn_triton.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/hf_prefixlm_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/hf_prefixlm_converter.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/meta_init_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/meta_init_context.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/modeling_mpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/modeling_mpt.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/norm.py -------------------------------------------------------------------------------- /llava/model/language_model/mpt/param_init_fns.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/language_model/mpt/param_init_fns.py -------------------------------------------------------------------------------- /llava/model/llava_arch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/llava_arch.py -------------------------------------------------------------------------------- /llava/model/make_delta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/make_delta.py -------------------------------------------------------------------------------- /llava/model/merger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/merger.py -------------------------------------------------------------------------------- /llava/model/multimodal_encoder/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/multimodal_encoder/builder.py -------------------------------------------------------------------------------- /llava/model/multimodal_encoder/clip_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/multimodal_encoder/clip_encoder.py -------------------------------------------------------------------------------- /llava/model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/model/utils.py -------------------------------------------------------------------------------- /llava/serve/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /llava/serve/cli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/cli.py -------------------------------------------------------------------------------- /llava/serve/controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/controller.py -------------------------------------------------------------------------------- /llava/serve/examples/ROCO_00027.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/examples/ROCO_00027.jpg -------------------------------------------------------------------------------- /llava/serve/examples/ROCO_00278.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/examples/ROCO_00278.jpg -------------------------------------------------------------------------------- /llava/serve/examples/ROCO_01259.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/examples/ROCO_01259.jpg -------------------------------------------------------------------------------- /llava/serve/examples/ROCO_01747.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/examples/ROCO_01747.jpg -------------------------------------------------------------------------------- /llava/serve/examples/ROCO_03257.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/examples/ROCO_03257.jpg -------------------------------------------------------------------------------- /llava/serve/gradio_web_server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/gradio_web_server.py -------------------------------------------------------------------------------- /llava/serve/model_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/model_worker.py -------------------------------------------------------------------------------- /llava/serve/register_worker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/register_worker.py -------------------------------------------------------------------------------- /llava/serve/test_message.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/serve/test_message.py -------------------------------------------------------------------------------- /llava/train/llama_flash_attn_monkey_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/train/llama_flash_attn_monkey_patch.py -------------------------------------------------------------------------------- /llava/train/llava_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/train/llava_trainer.py -------------------------------------------------------------------------------- /llava/train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/train/train.py -------------------------------------------------------------------------------- /llava/train/train_mem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/train/train_mem.py -------------------------------------------------------------------------------- /llava/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llava/utils.py -------------------------------------------------------------------------------- /llmenv.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/llmenv.yml -------------------------------------------------------------------------------- /playground/data/coco2014_val_gpt4_qa_30x3.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/coco2014_val_gpt4_qa_30x3.jsonl -------------------------------------------------------------------------------- /playground/data/coco2014_val_qa_eval/qa90_gpt4_answer.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/coco2014_val_qa_eval/qa90_gpt4_answer.jsonl -------------------------------------------------------------------------------- /playground/data/coco2014_val_qa_eval/qa90_questions.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/coco2014_val_qa_eval/qa90_questions.jsonl -------------------------------------------------------------------------------- /playground/data/prompts/complex_reasoning/000_caps.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/complex_reasoning/000_caps.txt -------------------------------------------------------------------------------- /playground/data/prompts/complex_reasoning/000_conv.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/complex_reasoning/000_conv.txt -------------------------------------------------------------------------------- /playground/data/prompts/complex_reasoning/001_caps.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/complex_reasoning/001_caps.txt -------------------------------------------------------------------------------- /playground/data/prompts/complex_reasoning/001_conv.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/complex_reasoning/001_conv.txt -------------------------------------------------------------------------------- /playground/data/prompts/complex_reasoning/002_caps.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/complex_reasoning/002_caps.txt -------------------------------------------------------------------------------- /playground/data/prompts/complex_reasoning/002_conv.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/complex_reasoning/002_conv.txt -------------------------------------------------------------------------------- /playground/data/prompts/complex_reasoning/system_message.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/complex_reasoning/system_message.txt -------------------------------------------------------------------------------- /playground/data/prompts/conversation/000_caps.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/conversation/000_caps.txt -------------------------------------------------------------------------------- /playground/data/prompts/conversation/000_conv.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/conversation/000_conv.txt -------------------------------------------------------------------------------- /playground/data/prompts/conversation/001_caps.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/conversation/001_caps.txt -------------------------------------------------------------------------------- /playground/data/prompts/conversation/001_conv.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/conversation/001_conv.txt -------------------------------------------------------------------------------- /playground/data/prompts/conversation/system_message.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/conversation/system_message.txt -------------------------------------------------------------------------------- /playground/data/prompts/detail_description/000_caps.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/detail_description/000_caps.txt -------------------------------------------------------------------------------- /playground/data/prompts/detail_description/000_conv.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/detail_description/000_conv.txt -------------------------------------------------------------------------------- /playground/data/prompts/detail_description/001_caps.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/detail_description/001_caps.txt -------------------------------------------------------------------------------- /playground/data/prompts/detail_description/001_conv.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/detail_description/001_conv.txt -------------------------------------------------------------------------------- /playground/data/prompts/detail_description/002_caps.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/detail_description/002_caps.txt -------------------------------------------------------------------------------- /playground/data/prompts/detail_description/002_conv.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/detail_description/002_conv.txt -------------------------------------------------------------------------------- /playground/data/prompts/detail_description/system_message.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/playground/data/prompts/detail_description/system_message.txt -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/pyproject.toml -------------------------------------------------------------------------------- /scripts/convert_sqa_to_llava.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/convert_sqa_to_llava.py -------------------------------------------------------------------------------- /scripts/convert_sqa_to_llava_base_prompt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/convert_sqa_to_llava_base_prompt.py -------------------------------------------------------------------------------- /scripts/copy_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/copy_model.py -------------------------------------------------------------------------------- /scripts/extract_mm_projector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/extract_mm_projector.py -------------------------------------------------------------------------------- /scripts/filter_missing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/filter_missing.py -------------------------------------------------------------------------------- /scripts/finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/finetune.sh -------------------------------------------------------------------------------- /scripts/finetune_lora.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/finetune_lora.sh -------------------------------------------------------------------------------- /scripts/finetune_qlora.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/finetune_qlora.sh -------------------------------------------------------------------------------- /scripts/inference.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/inference.sh -------------------------------------------------------------------------------- /scripts/load_from_hf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/load_from_hf.py -------------------------------------------------------------------------------- /scripts/massage_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/massage_data.py -------------------------------------------------------------------------------- /scripts/merge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/merge.sh -------------------------------------------------------------------------------- /scripts/merge_lora_weights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/merge_lora_weights.py -------------------------------------------------------------------------------- /scripts/misc/inspect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/misc/inspect.py -------------------------------------------------------------------------------- /scripts/pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/pretrain.sh -------------------------------------------------------------------------------- /scripts/quantise.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/quantise.ipynb -------------------------------------------------------------------------------- /scripts/quantise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/quantise.py -------------------------------------------------------------------------------- /scripts/sqa_eval_batch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/sqa_eval_batch.sh -------------------------------------------------------------------------------- /scripts/sqa_eval_gather.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/sqa_eval_gather.sh -------------------------------------------------------------------------------- /scripts/unlink.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/unlink.sh -------------------------------------------------------------------------------- /scripts/upload_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/upload_dataset.py -------------------------------------------------------------------------------- /scripts/upload_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/upload_lora.py -------------------------------------------------------------------------------- /scripts/zero2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/zero2.json -------------------------------------------------------------------------------- /scripts/zero3.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/zero3.json -------------------------------------------------------------------------------- /scripts/zero3_offload.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/photomz/BabyDoctor/HEAD/scripts/zero3_offload.json --------------------------------------------------------------------------------