| 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}") | |