Temporarily fix PyTorch version to 2.0.1 as the latest PyTorch requir…

…es CUDA 12.1 by default.
dudulightricks · Oct 10, 2023 · fda0665 · fda0665
1 parent e854a2b
commit fda0665
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,7 +15,7 @@ classifiers = [
 dependencies = [
     "einops", "fastapi", "gradio==3.35.2", "markdown2[all]", "numpy",
     "requests", "sentencepiece", "tokenizers>=0.12.1",
-    "torch", "torchvision", "uvicorn", "wandb",
+    "torch==2.0.1", "torchvision==0.15.2", "uvicorn", "wandb",
     "shortuuid", "httpx==0.24.0",
     "deepspeed==0.9.5",
     "peft==0.4.0",