1Shanghai Artificial Intelligence Laboratory
🔥The technical report is coming soon!
- [2024.09.14] 🔥 Inference code and checkpoint are released.
Note: You may want to adjust the CUDA version according to your driver version.
conda create -n VchitectXL -y
conda activate VchitectXL
conda install python=3.11 pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia -ypip install -r requirements.txtFirst download the checkpoint.
test_file=$1
save_dir=$2
ckpt_path=$3
python inference.py --test_file "${test_file}" --save_dir "${save_dir}" --ckpt_path "${ckpt_path}"
This code is licensed under Apache-2.0. The framework is fully open for academic research and also allows free commercial usage.