diff --git a/pyproject.toml b/pyproject.toml index 54ddd87..b10212e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -29,7 +29,7 @@ dependencies = [ "pydantic==1.10.5", "code-tokenize==0.2.0", "openai==1.3.3", - "torch==1.12.0", + "torch==1.13.1", "transformers==4.30.0", "tiktoken==0.3.3", "pygtrie==2.5.0" diff --git a/requirements.txt b/requirements.txt index af6ca8f..83bdc3d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,4 +14,4 @@ pygtrie==2.5.0 openai==1.3.3 code-tokenize==0.2.0 --extra-index-url https://download.pytorch.org/whl/cu113 -torch==1.12.0+cu113 \ No newline at end of file +torch==1.13.1 \ No newline at end of file