github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update

add_xcomposer_testcase
zhulin1 2024-07-16 18:55:25 +08:00
parent 7a782d6947
commit f3b3e740a8
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/daily_tests.yaml vendored
View File

@ -41,7 +41,7 @@ jobs:
source activate internlm-model-latest
pip install torch==2.2.2 torchvision==0.17.2 --index-url https://download.pytorch.org/whl/cu118
pip install /mnt/petrelfs/qa-caif-cicd/resource/flash_attn-2.5.8+cu118torch2.2cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
pip install sentencepiece auto-gptq==0.6.0 lmdeploy[all] beautifulsoup4 decord
pip install sentencepiece auto-gptq==0.6.0 lmdeploy[all] beautifulsoup4 decord lxml
srun -p ${SLURM_PARTITION} --kill-on-bad-exit=1 --job-name=${GITHUB_RUN_ID}-${GITHUB_JOB} --gpus-per-task=2 pytest -s -v -m tmp --color=yes ./tests/test_hf_model.py
conda deactivate
- name: remove_env

2
tests/test_hf_model.py
View File

@ -394,7 +394,7 @@ class TestXcomposer2d5Model:
model.tokenizer = tokenizer
query = 'Generate the HTML code of this web image with Tailwind CSS.'
image = ['./examples/screenshot.jpg']
image = ['/mnt/petrelfs/qa-caif-cicd/github_runner/examples/screenshot.jpg']
with torch.autocast(device_type='cuda', dtype=torch.float16):
response = model.resume_2_webpage(query,
image,