Spaces:
Running
on
Zero
Running
on
Zero
File size: 557 Bytes
2ada650 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
#!/bin/bash
#SBATCH --partition=batch
#SBATCH --job-name=video_demo_llama2
#SBATCH --output=video_demo_llama2.out
#SBATCH --error=video_demo_llama2.err
#SBATCH --time=0-10:30:00
#SBATCH --mem=100G
#SBATCH --gres=gpu:a100:1
#SBATCH --nodes=1
# Choose the model to test
# Mistral
# ckpt="checkpoints/video_mistral_checkpoint_last.pth"
# config="test_configs/mistral_test_config.yaml"
# Llama2
ckpt="checkpoints/video_llama_checkpoint_last.pth"
config="test_configs/llama2_test_config.yaml"
python minigpt4_video_demo.py --cfg-path $config --ckpt $ckpt
|