Fix torch version warning

2 years ago · c50e3485d0
parent 6cd5c7a760
commit c50e3485d0
2 changed files with 2 additions and 2 deletions
--- a/requirements.txt
+++ b/requirements.txt
@ -1,5 +1,5 @@
 coremltools
 diffusers[torch]
-torch
+torch == 1.12.1
 transformers
 scipy
--- a/setup.py
+++ b/setup.py
@ -16,7 +16,7 @@ setup(
    install_requires=[
        "coremltools>=6.1",
        "diffusers[torch]",
-        "torch",
+        "torch==1.12.1",
        "transformers",
        "scipy",
        "numpy<1.24",