GitHub - Yuan-1028/UI-Transition

llava

执行脚本 openai_llava.py

python 3.10 cuda 11.8

torch==2.2.2
torchvision==0.17.2
transformers==4.43.4

pip install -r requirements.txt

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
.ipynb_checkpoints		.ipynb_checkpoints
logs		logs
.gitignore		.gitignore
README.md		README.md
Roboto-Bold.ttf		Roboto-Bold.ttf
Untitled.ipynb		Untitled.ipynb
api.py		api.py
build_dataset.py		build_dataset.py
ds_config.yaml		ds_config.yaml
environment.yml		environment.yml
hfd.sh		hfd.sh
infer.py		infer.py
openai_llava.py		openai_llava.py
peft.infer.py		peft.infer.py
process_data.py		process_data.py
requirements.txt		requirements.txt
sft.py		sft.py