From c50e3485d0edeadc8972d209d91c36a3d290242d Mon Sep 17 00:00:00 2001
From: SNQ-2001 <taishin05109@gmail.com>
Date: Fri, 27 Jan 2023 23:57:48 +0900
Subject: [PATCH] Fix torch version warning

---
 requirements.txt | 2 +-
 setup.py         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index d48a002..8cee717 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 coremltools
 diffusers[torch]
-torch
+torch == 1.12.1
 transformers
 scipy
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 69c0e39..9bb850f 100644
--- 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",