diff --git a/requirements.txt b/requirements.txt index f45766b8..13b4ffef 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ -torch -torchvision +torch==2.5.1 +torchvision==0.20.1 safetensors git+https://github.com/huggingface/diffusers.git transformers