Skip to content

shiym2000/getting-started-with-lvlms

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Getting Started with Large Vision-Lauguage Models (LVLMs)

代码结构

监督微调(SFT)

src/train.py:解析参数-构建模型-构建数据集-训练模型-保存模型

推理(单轮对话)

src/inference.py:解析参数-加载模型-构建数据集-模型推理-保存结果

推理(多轮对话)

src/inference_multiturn.py:解析参数-加载模型-构造数据-模型推理-保存结果

开发日志

  • src/dataset.py
  • src/model.py
  • src/train.py
  • 基于DeepSpeed并行训练(ZeRO-2/ZeRO-3)
  • src/inference.py
  • src/inference_multiturn.py
  • LoRA微调
  • QLoRA微调
  • 基于vLLM加速推理

About

Getting Started with Large Vision-Lauguage Models (LVLMs)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published