awoo

Signed-off-by: Balazs Horvath <[email protected]>

dataset_tools/Create Empty caption Files for Images.ipynb

@@ -1,8 +1,17 @@
 {
  "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Create Empty `.txt` Files For Images\n",
+    "\n",
+    "---"
+   ]
+  },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {},
    "outputs": [
     {
@@ -11,16 +20,43 @@
      "text": [
       "Caption files created successfully.\n"
      ]
-    },
+    }
+   ],
+   "source": [
+    "from pathlib import Path\n",
+    "\n",
+    "def create_caption_file(directory):\n",
+    "    for image_path in Path(directory).rglob('*'):\n",
+    "        if image_path.suffix.lower() in ['.jpg', '.jpeg', '.png']:\n",
+    "            caption_file = image_path.with_suffix('.txt')\n",
+    "            if not caption_file.exists():\n",
+    "                caption_file.touch()\n",
+    "\n",
+    "if __name__ == \"__main__\":\n",
+    "    directory = r'E:\\training_dir_staging'\n",
+    "    create_caption_file(directory)\n",
+    "    print(\"Caption files created successfully.\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Create Empty `.caption` Files For Images\n",
+    "\n",
+    "---"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 1,
+   "metadata": {},
+   "outputs": [
     {
-     "ename": "",
-     "evalue": "",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[1;31mThe Kernel crashed while executing code in the current cell or a previous cell. \n",
-      "\u001b[1;31mPlease review the code in the cell(s) to identify a possible cause of the failure. \n",
-      "\u001b[1;31mClick <a href='https://aka.ms/vscodeJupyterKernelCrash'>here</a> for more info. \n",
-      "\u001b[1;31mView Jupyter <a href='command:jupyter.viewOutput'>log</a> for further details."
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Caption files created successfully.\n"
      ]
     }
    ],

tests/earbou-v1e400/earbou-v1e400.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0f29ae81f39061a67925db77bec4b7a459fc9a23c35f651727aa079b5b1ce823
+size 240580432

tests/earbou-v2e400/earbou-v2e400.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:753faca3409886c93bfc4bc9de9173dc655dbddc2c2f127e40a764c341e8f4c8
+size 240587808