Created using Colaboratory

Train_Colab.ipynb

@@ -1,18 +1,16 @@
 {
   "cells": [
     {
-      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
-        "colab_type": "text",
-        "id": "view-in-github"
+        "id": "view-in-github",
+        "colab_type": "text"
       },
       "source": [
-        "<a href=\"https://colab.research.google.com/github/victorchall/EveryDream2trainer/blob/main/Train_Colab.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
+        "<a href=\"https://colab.research.google.com/github/nawnie/EveryDream2trainer/blob/main/Train_Colab.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
       ]
     },
     {
-      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "blaLMSbkPHhG"
@@ -147,7 +145,6 @@
       ]
     },
     {
-      "attachments": {},
       "cell_type": "markdown",
       "metadata": {
         "id": "nEzuEYH0536C"
@@ -389,7 +386,8 @@
       "cell_type": "code",
       "execution_count": null,
       "metadata": {
-        "id": "Iuoa_1B9jRGU"
+        "id": "Iuoa_1B9jRGU",
+        "cellView": "form"
       },
       "outputs": [],
       "source": [
@@ -397,6 +395,7 @@
         "#@markdown * Edit train.json to setup your paramaters\n",
         "#@markdown * Edit chain0.json to make use of chaining\n",
         "#@markdown * make sure to check each confguration you will need 1 Json per chain length 3 are provided\n",
+        "#@markdown * make sure your .Json contain the line Notebook: true\n",
         "\n",
         "\n",
         "%cd /content/EveryDream2trainer\n",
@@ -410,13 +409,42 @@
         "  l -= 1\n",
         "  I =+ 1"
       ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "#@title Test your Diffusers\n",
+        "#@markdown Path to the diffusers that was trained\n",
+        "diffusers_path=\"\" #@param{type:\"string\"}\n",
+        "DF=diffusers_path\n",
+        "PROMPT= \"a photo of an astronaut on the moon\"#@param{type:\"string\"}\n",
+        "Resolution = 512 #@param {type:\"slider\", min:256, max:1024, step:32}\n",
+        "Seed= -1 #@param{type:\"integer\"}\n",
+        "Steps = 30 #@param {type:\"slider\", min:10, max:50, step:1}\n",
+        "cfg = 7 #@param {type:\"slider\", min:1, max:15, step:0.5}\n",
+        "\n",
+        "\n",
+        "!python /content/EveryDream2trainer/scripts/txt2img.py   \\\n",
+        " --diffusers_path \"$DF\" \\\n",
+        " --resolution $Resolution \\\n",
+        " --seed $Seed \\\n",
+        " --prompt \"$PROMPT\" \\\n",
+        " --steps $Steps \\\n",
+        " --cfg_scale $cfg "
+      ],
+      "metadata": {
+        "cellView": "form",
+        "id": "8HmIWtODuE6p"
+      },
+      "execution_count": null,
+      "outputs": []
     }
   ],
   "metadata": {
     "accelerator": "GPU",
     "colab": {
-      "include_colab_link": true,
-      "provenance": []
+      "provenance": [],
+      "include_colab_link": true
     },
     "gpuClass": "standard",
     "kernelspec": {
@@ -436,4 +464,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
+}