.

check_hf.sh

@@ -1,11 +1,16 @@
 python - <<'PY'
-import transformers, accelerate, datasets, safetensors, sentencepiece, peft, bitsandbytes
+import transformers, accelerate, datasets, safetensors, sentencepiece, peft
+from importlib.metadata import version, PackageNotFoundError
+
 print("Transformers:", transformers.__version__)
 print("Accelerate:", accelerate.__version__)
 print("Datasets:", datasets.__version__)
 print("Safetensors:", safetensors.__version__)
 print("SentencePiece:", sentencepiece.__version__)
 print("PEFT:", peft.__version__)
-print("BitsAndBytes:", bitsandbytes.__version__)
-PY
 
+try:
+    print("BitsAndBytes:", version("bitsandbytes"))
+except PackageNotFoundError:
+    print("BitsAndBytes: N/A")
+PY

requirements-hf.txt

@@ -1,5 +1,5 @@
 peft>=0.11
-bitsandbytes>=0.43
+bitsandbytes==0.41.1
 # 可选：
 # xformers==0.0.22.post3
 # flash-attn