he library environment dependencies

chujiezheng / LLM-Safeguard

Official repository for ICML 2024 paper "On Prompt-Driven Safeguarding for Large Language Models"

https://arxiv.org/abs/2401.18018

65 stars 6 forks source link

Closed NieSYsc20 closed 3 months ago

NieSYsc20 commented 3 months ago

Could you please provide the library environment dependencies for this project?

chujiezheng commented 3 months ago

Sorry that I no longer have access to the GPU machine. Based on my memory, here are some key package versions:

python=3.10
pytorch=2.1.1
cuda=11.8
transformers~=4.36