Upload sd_token_similarity_calculator.ipynb

sd_token_similarity_calculator.ipynb

@@ -318,13 +318,7 @@
       "source": [
         "# @title 📝 Get Prompt text_encoding similarity to the pre-calc. text_encodings\n",
         "prompt = \" a fast car on the road \" # @param {\"type\":\"string\",\"placeholder\":\"Write a prompt\"}\n",
-        "list_size = 100 # @param {type:'number'}\n",
-        "start_at_index = 0 # @param {type:'number'}\n",
-        "print_Similarity = True # @param {type:\"boolean\"}\n",
-        "print_Suffix = True # @param {type:\"boolean\"}\n",
-        "print_Prefix = True # @param {type:\"boolean\"}\n",
-        "print_Descriptions = True # @param {type:\"boolean\"}\n",
-        "compact_Output = False # @param {type:\"boolean\"}\n",
+        "\n",
         "\n",
         "from transformers import AutoTokenizer\n",
         "tokenizer = AutoTokenizer.from_pretrained(\"openai/clip-vit-large-patch14\", clean_up_tokenization_spaces = False)\n",
@@ -379,8 +373,26 @@
         "#------#\n",
         "suffix_sorted, suffix_indices = torch.sort(dots,dim=0 , descending=True)\n",
         "#------#\n",
+        "\n"
+      ],
+      "metadata": {
+        "id": "xc-PbIYF428y"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# @title 📝 Print the results\n",
+        "list_size = 100 # @param {type:'number'}\n",
+        "start_at_index = 0 # @param {type:'number'}\n",
+        "print_Similarity = True # @param {type:\"boolean\"}\n",
+        "print_Suffix = True # @param {type:\"boolean\"}\n",
+        "print_Prefix = True # @param {type:\"boolean\"}\n",
+        "print_Descriptions = True # @param {type:\"boolean\"}\n",
+        "compact_Output = False # @param {type:\"boolean\"}\n",
         "\n",
-        "#Print the results\n",
         "# title Show the 100 most similiar suffix and prefix text-encodings to the text encoding\n",
         "RANGE = list_size\n",
         "_suffixes = '{'\n",
@@ -444,11 +456,10 @@
         "  if(compact_Output):\n",
         "    print((prefixes + _suffixes).replace('}{', '|'))\n",
         "  else:\n",
-        "    print(prefixes)\n",
-        "\n"
+        "    print(prefixes)"
       ],
       "metadata": {
-        "id": "xc-PbIYF428y"
+        "id": "_vnVbxcFf7WV"
       },
       "execution_count": null,
       "outputs": []
@@ -512,7 +523,7 @@
           "height": 1000
         }
       },
-      "execution_count": 4,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "display_data",
@@ -593,10 +604,26 @@
         "  d.close() #close the file\n",
         "#------#\n",
         "suffix_sorted, suffix_indices = torch.sort(dots,dim=0 , descending=True)\n",
-        "#------#\n",
-        "\n",
+        "#------#"
+      ],
+      "metadata": {
+        "id": "rebogpoyOG8k"
+      },
+      "execution_count": null,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# @title 🖼️ Print the results\n",
+        "list_size = 100 # @param {type:'number'}\n",
+        "start_at_index = 0 # @param {type:'number'}\n",
+        "print_Similarity = True # @param {type:\"boolean\"}\n",
+        "print_Suffix = True # @param {type:\"boolean\"}\n",
+        "print_Prefix = True # @param {type:\"boolean\"}\n",
+        "print_Descriptions = True # @param {type:\"boolean\"}\n",
+        "compact_Output = False # @param {type:\"boolean\"}\n",
         "\n",
-        "#Print the results\n",
         "# title Show the 100 most similiar suffix and prefix text-encodings to the text encoding\n",
         "RANGE = list_size\n",
         "_suffixes = '{'\n",
@@ -619,21 +646,21 @@
         "  name = ahead + get_suffix(id) + behind\n",
         "  if(get_suffix(id) == ' '): name = ahead + f'{id}' + behind\n",
         "  _suffixes = _suffixes + name + '|'\n",
-        "  _sims = _sims + f'{round(sim,2)} %' + '|'\n",
+        "  _sims = _sims + f'{round(sim*100,2)} %' + '|'\n",
         "#------#\n",
         "_suffixes = (_suffixes + '}').replace('|}', '}')\n",
         "_sims = (_sims + '}').replace('|}', '}')\n",
         "#------#\n",
         "\n",
+        "\n",
         "suffixes = _suffixes\n",
         "sims = _sims\n",
-        "\n",
         "if(not print_Suffix): suffixes = ''\n",
         "if(not print_Similarity): sims = ''\n",
         "\n",
         "if(not compact_Output):\n",
         "  if(print_Descriptions):\n",
-        "    print(f'The {start_at_index}-{start_at_index + RANGE} most similiar suffix items to prompt : ' + suffixes)\n",
+        "    print(f'The {start_at_index}-{start_at_index + RANGE} most similiar suffix items to image : ' + suffixes)\n",
         "    print(f'The {start_at_index}-{start_at_index + RANGE} similarity % for suffix items : ' + sims)\n",
         "    print('')\n",
         "  else:\n",
@@ -655,15 +682,15 @@
         "if(not print_Prefix): prefixes = ''\n",
         "\n",
         "if(print_Descriptions):\n",
-        "  print(f'The {start_at_index}-{start_at_index + RANGE} most similiar prefixes to prompt : ' + prefixes)\n",
+        "  print(f'The {start_at_index}-{start_at_index + RANGE} most similiar prefixes to image : ' + prefixes)\n",
         "else:\n",
         "  if(compact_Output):\n",
         "    print((prefixes + _suffixes).replace('}{', '|'))\n",
         "  else:\n",
-        "    print(prefixes)\n"
+        "    print(prefixes)"
       ],
       "metadata": {
-        "id": "rebogpoyOG8k"
+        "id": "JkzncP8SgKtS"
       },
       "execution_count": null,
       "outputs": []