train_env_prepare/check_hf.sh

17 lines
541 B
Bash

python - <<'PY'
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__)
try:
print("BitsAndBytes:", version("bitsandbytes"))
except PackageNotFoundError:
print("BitsAndBytes: N/A")
PY