File size: 938 Bytes
b7ff7e3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
tests = {
"torch": "import torch; v=torch.__version__",
"transformers": "import transformers; v=transformers.__version__",
"Qwen2.5VL_model": "from transformers import Qwen2_5_VLForConditionalGeneration; v='OK'",
"AutoProcessor": "from transformers import AutoProcessor; v='OK'",
"accelerate": "import accelerate; v=accelerate.__version__",
"deepspeed": "import deepspeed; v=deepspeed.__version__",
"peft": "import peft; v=peft.__version__",
"flash_attn": "import flash_attn; v=flash_attn.__version__",
"ms_swift": "from swift.llm import sft_main; v='OK'",
"PIL": "from PIL import Image; v='OK'",
"qwen_vl_utils": "import qwen_vl_utils; v='OK'",
"CUDA": "import torch; v=str(torch.cuda.device_count())+' GPUs'",
}
for name, code in tests.items():
try:
v = None
exec(code)
print(f"OK {name}: {v}")
except Exception as e:
print(f"FAIL {name}: {e}")
|