Fixing inputs with spaces not making their way into main.py

Author: djbielejeski

dreambooth_colab_joepenna.ipynb

@@ -15,6 +15,19 @@
   }
  },
  "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "outputs": [],
+   "source": [
+    "#@title Load repo (if needed)\n",
+    "!git clone https://github.com/JoePenna/Dreambooth-Stable-Diffusion\n",
+    "%cd Dreambooth-Stable-Diffusion"
+   ],
+   "metadata": {
+    "collapsed": false
+   }
+  },
   {
    "cell_type": "code",
    "execution_count": null,
@@ -24,11 +37,6 @@
    "outputs": [],
    "source": [
     "#@title BUILD ENV\n",
-    "!git clone https://github.com/JoePenna/Dreambooth-Stable-Diffusion\n",
-    "%cd Dreambooth-Stable-Diffusion\n",
-    "\n",
-    "!git checkout feature/google-colab-notebook\n",
-    "\n",
     "!pip install omegaconf\n",
     "!pip install einops\n",
     "!pip install pytorch-lightning==1.6.5\n",
@@ -50,7 +58,7 @@
     "!pip install ipywidgets==7.7.1\n",
     "\n",
     "import os\n",
-    "os._exit(00)\n"
+    "os._exit(00)"
    ]
   },
   {
@@ -188,13 +196,13 @@
     " --base configs/stable-diffusion/v1-finetune_unfrozen.yaml \\\n",
     " -t \\\n",
     " --actual_resume \"model.ckpt\" \\\n",
-    " --reg_data_root {reg_data_root} \\\n",
-    " -n {project_name} \\\n",
+    " --reg_data_root \"{reg_data_root}\" \\\n",
+    " -n \"{project_name}\" \\\n",
     " --gpus 0, \\\n",
     " --data_root \"/content/Dreambooth-Stable-Diffusion/training_images\" \\\n",
     " --max_training_steps {max_training_steps} \\\n",
-    " --class_word {class_word} \\\n",
-    " --token {token} \\\n",
+    " --class_word \"{class_word}\" \\\n",
+    " --token \"{token}\" \\\n",
     " --no-test"
    ],
    "metadata": {

dreambooth_runpod_joepenna.ipynb

@@ -349,13 +349,13 @@
     " --base configs/stable-diffusion/v1-finetune_unfrozen.yaml \\\n",
     " -t \\\n",
     " --actual_resume \"model.ckpt\" \\\n",
-    " --reg_data_root {reg_data_root} \\\n",
-    " -n {project_name} \\\n",
+    " --reg_data_root \"{reg_data_root}\" \\\n",
+    " -n \"{project_name}\" \\\n",
     " --gpus 0, \\\n",
     " --data_root \"/workspace/Dreambooth-Stable-Diffusion/training_images\" \\\n",
     " --max_training_steps {max_training_steps} \\\n",
-    " --class_word {class_word} \\\n",
-    " --token {token} \\\n",
+    " --class_word \"{class_word}\" \\\n",
+    " --token \"{token}\" \\\n",
     " --no-test"
    ]
   },