fix(install): update install.sh

README.md

@@ -31,12 +31,14 @@ These instructions will help you run **quick inference locally**.
 Run the following command in the **top-level directory**:
 
 ```shell
+chmod +x script/install.sh
 bash script/install.sh
 ```
 
 #### 🐳 Option 2: Using Docker
 
 ```shell
+chmod +x script/run_svgdreamer_docker.sh
 sudo bash script/run_svgdreamer_docker.sh
 ```
 

script/install.sh

@@ -1,15 +1,23 @@
 #!/bin/bash
+
+set -e
+
+# Conda setup and environment creation
 eval "$(conda shell.bash hook)"
 
 conda create --name svgrender python=3.10 --yes
 conda activate svgrender
-
 echo "The conda environment was successfully created"
 
+# Install PyTorch and related libraries
 conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch --yes
-
 echo "Pytorch installation is complete. version: 1.12.1"
 
+# Install xformers
+conda install xformers -c xformers --yes
+echo "xformers installation is complete."
+
+# Install common Python dependencies
 pip install hydra-core omegaconf
 pip install freetype-py shapely svgutils
 pip install opencv-python scikit-image matplotlib visdom wandb BeautifulSoup4
@@ -17,31 +25,34 @@ pip install triton numba
 pip install numpy scipy scikit-fmm einops timm fairscale==0.4.13
 pip install accelerate transformers safetensors datasets
 pip install easydict scikit-learn pytorch_lightning==2.1.0 webdataset
-
 echo "The basic dependency library is installed."
 
+# Additional utility libraries
 pip install ftfy regex tqdm
 pip install git+https://github.com/openai/CLIP.git
-
 echo "CLIP installation is complete."
 
+# Install diffusers
 pip install diffusers==0.20.2
-
 echo "Diffusers installation is complete. version: 0.20.2"
-# if xformers doesnt install properly with conda try installing with pip using the code below
-# pip install --pre -U xformers
-conda install xformers -c xformers --yes
-
-echo "xformers installation is complete."
 
+# Clone and set up DiffVG, handling dependencies on Ubuntu
 git clone https://github.com/BachiLi/diffvg.git
 cd diffvg
 git submodule update --init --recursive
+
+# Install system dependencies for Ubuntu (to avoid potential issues)
+echo "Installing system dependencies for DiffVG..."
+sudo apt update
+sudo apt install -y cmake ffmpeg build-essential libjpeg-dev libpng-dev libtiff-dev
+
 conda install -y -c anaconda cmake
 conda install -y -c conda-forge ffmpeg
 pip install svgwrite svgpathtools cssutils torch-tools
-python setup.py install
 
+# Install DiffVG
+python setup.py install
 echo "DiffVG installation is complete."
 
-echo "the running environment has been successfully installed!!!"
+# Final confirmation
+echo "The running environment has been successfully installed!!!"