Update requirements.txt to resolve installation issues

README.md

@@ -45,6 +45,7 @@
 - Install the required dependencies:
 
     ```bash
+    pip install pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu118
     pip install -r requirements.txt
     ```
 

install.bat

@@ -0,0 +1,24 @@
+@echo off
+REM Check for Python and exit if not found
+python --version >nul 2>&1
+if %errorlevel% neq 0 (
+    echo Python is not installed. Please install Python and retry.
+    exit /b
+)
+
+REM Create a virtual environment
+python -m venv venv
+
+REM Activate the virtual environment
+call venv\Scripts\activate.bat
+
+REM Upgrade pip before `pip install`
+python -m pip install --upgrade pip
+
+pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu118
+
+REM Install other requirements
+pip install -r requirements.txt
+
+echo Installation completed.
+pause

install.sh

@@ -0,0 +1,23 @@
+#!/bin/bash
+set -e
+cd "$(dirname "$0")"
+# Check for Python and exit if not found
+if ! [ -x "$(command -v python)" ]; then
+  echo 'Error: python is not installed.' >&2
+  exit 1
+fi
+# Create a virtual environment
+if [ ! -d "venv" ]; then
+  python -m venv venv
+fi
+
+# Upgrade pip before `pip install`
+python -m pip install --upgrade pip
+
+# Install the custom diffusers version from GitHub
+./venv/bin/pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu118
+
+# Install other requirements
+./venv/bin/pip install -r requirements.txt
+
+echo "Installation completed."

requirements.txt

@@ -1,44 +1,6 @@
-accelerate==0.24.1
-certifi==2023.11.17
-charset-normalizer==3.3.2
 diffusers==0.19.3
-filelock==3.13.1
-fsspec==2023.10.0
-huggingface-hub==0.19.4
-idna==3.4
-importlib-metadata==6.8.0
-Jinja2==3.1.2
-MarkupSafe==2.1.3
-mpmath==1.3.0
-networkx==3.2.1
-numpy==1.26.2
-nvidia-cublas-cu12==12.1.3.1
-nvidia-cuda-cupti-cu12==12.1.105
-nvidia-cuda-nvrtc-cu12==12.1.105
-nvidia-cuda-runtime-cu12==12.1.105
-nvidia-cudnn-cu12==8.9.2.26
-nvidia-cufft-cu12==11.0.2.54
-nvidia-curand-cu12==10.3.2.106
-nvidia-cusolver-cu12==11.4.5.107
-nvidia-cusparse-cu12==12.1.0.106
-nvidia-nccl-cu12==2.18.1
-nvidia-nvjitlink-cu12==12.3.101
-nvidia-nvtx-cu12==12.1.105
-packaging==23.2
-Pillow==10.1.0
-psutil==5.9.6
-PyYAML==6.0.1
-regex==2023.10.3
-requests==2.31.0
-safetensors==0.4.0
-sentencepiece==0.1.99
-sympy==1.12
-tokenizers==0.15.0
-torch==2.1.1
-torchvision==0.16.1
-tqdm==4.66.1
-transformers==4.35.2
-triton==2.1.0
-typing_extensions==4.8.0
-urllib3==2.1.0
-zipp==3.17.0
+transformers>=4.35.2
+torch>=2.1.2+cu118
+torchvision>=0.16.2+cu118
+accelerate>=0.27.2
+sentencepiece # Required for DeepFloyd/IF