cuda update 12.1

inkn · Dec 30, 2023 · fa8d196 · fa8d196
1 parent 3a485af
commit fa8d196
Show file tree

Hide file tree

Showing 6 changed files with 44 additions and 1,452 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -132,7 +132,7 @@ termcolor==2.3.0
 threadpoolctl==3.2.0
 tiktoken==0.3.3
 tokenizers==0.14.1
-torch==2.1.1+cu118
+torch  --index-url https://download.pytorch.org/whl/cu121
 tqdm==4.66.1
 typer==0.3.2
 typing_extensions==4.8.0