diff --git a/pyproject.toml b/pyproject.toml index 2f53f57..d416cd4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -30,7 +30,7 @@ dependencies = [ "code-tokenize==0.2.0", "code-ast @ git+https://github.com/cedricrupb/code_ast@982940d04b1d721e5ac9a97d433f36d1fb47e8e0", "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 c5cc3c2..77b8bd3 100644 --- a/requirements.txt +++ b/requirements.txt @@ -15,4 +15,4 @@ openai==1.3.3 code-tokenize==0.2.0 code-ast @ git+https://github.com/cedricrupb/code_ast@982940d04b1d721e5ac9a97d433f36d1fb47e8e0 --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