Upload root_scripts/test_env.py with huggingface_hub

root_scripts/test_env.py

@@ -0,0 +1,22 @@
+
+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}")