From dcc6a146e660e2610b7430c31db7b20026611254 Mon Sep 17 00:00:00 2001
From: hoangtung386 <levuhoangtung1542003@gmail.com>
Date: Thu, 25 Jun 2026 18:06:56 +0700
Subject: [PATCH] Fix environment installation: Python and PyTorch CUDA
 versions

---
 README.md        | 2 +-
 requirements.txt | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 083dc2a..27dfed3 100644
--- a/README.md
+++ b/README.md
@@ -114,7 +114,7 @@ The masks of the evaluation set are shared in [Google Drive](https://drive.googl
 * See 'requirements.txt' for detailed Python libraries required
 
 ```bash
-conda create -n moebius python=3.14.4
+conda create -n moebius python=3.12
 conda activate moebius
 # cd /xx/xx/Moebius
 pip install -r requirements.txt
diff --git a/requirements.txt b/requirements.txt
index 9e5aad8..ee47d4e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
-torch==2.7.1+cu130
-torchvision
+--extra-index-url https://download.pytorch.org/whl/cu128
+torch==2.7.1+cu128
+torchvision==0.22.1+cu128
 diffusers==0.38.0
 transformers
 accelerate==1.14.0