diff --git a/check_hf.sh b/check_hf.sh index 12833e7..5498ebf 100644 --- a/check_hf.sh +++ b/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 diff --git a/requirements-hf.txt b/requirements-hf.txt index ba5987e..c6c6bf4 100644 --- a/requirements-hf.txt +++ b/requirements-hf.txt @@ -1,5 +1,5 @@ peft>=0.11 -bitsandbytes>=0.43 +bitsandbytes==0.41.1 # 可选: # xformers==0.0.22.post3 # flash-attn \ No newline at end of file