search

deadbits / vigil-llm

⚡ Vigil ⚡ Detect prompt injections, jailbreaks, and other potentially risky Large Language Model (LLM) inputs
https://vigil.deadbits.ai/
Apache License 2.0
304 stars 35 forks source link

Pin torch versions in Dockerfile #71

Closed deadbits closed 10 months ago

deadbits commented 10 months ago

Addresses https://github.com/deadbits/vigil-llm/issues/69

Torch and torchvision python versions in the Dockerfile. Also removed torchaudio as it is not needed

deadbits commented 10 months ago

Damn, need to pull out the --no-recommends first so I can merge that in the existing PR