diff --git a/models/requirements.txt b/models/requirements.txt index ecd3ed65..b9fe9d5e 100644 --- a/models/requirements.txt +++ b/models/requirements.txt @@ -1,4 +1,4 @@ numpy -torch +torch >= 1.10 safetensors -pytorch_lightning \ No newline at end of file +pytorch_lightning