Upload sd_token_similarity_calculator.ipynb

sd_token_similarity_calculator.ipynb

@@ -122,7 +122,8 @@
         "outputId": "033c251a-2043-40e7-9500-4da870ffa7fd",
         "colab": {
           "base_uri": "https://localhost:8080/"
-        }
+        },
+        "cellView": "form"
       },
       "execution_count": null,
       "outputs": [
@@ -387,7 +388,7 @@
         "start_search_at_index = 0 # @param {type:\"slider\", min:0, max: 49407, step:100}\n",
         "# @markdown The lower the start_index, the more similiar the sampled tokens will be to the target token assigned in the '⚡ Get similiar tokens' cell\". If the cell was not run, then it will use tokens ordered by similarity to the \"girl\\</w>\" token\n",
         "start_search_at_ID = start_search_at_index\n",
-        "search_range = 1000 # @param {type:\"slider\", min:10, max: 2000, step:10}\n",
+        "search_range = 1000 # @param {type:\"slider\", min:100, max:49407, step:100}\n",
         "\n",
         "samples_per_iter = 10 # @param {type:\"slider\", min:10, max: 100, step:10}\n",
         "\n",
@@ -442,7 +443,7 @@
         "  #-----#\n",
         "\n",
         "  for index in range(samples_per_iter):\n",
-        "    _start = START + iter*RANGE\n",
+        "    _start = START\n",
         "    id_C = random.randint(_start , _start + RANGE)\n",
         "    name_C = db_vocab[f'{id_C}']\n",
         "    is_Prefix = 0\n",
@@ -517,7 +518,7 @@
         "  sorted, indices = torch.sort(dots,dim=0 , descending=True)\n",
         "  # @markdown ----------\n",
         "  # @markdown # Print options\n",
-        "  list_size = 100 # @param {type:'number'}\n",
+        "  list_size = 100 # param {type:'number'}\n",
         "  print_ID = False # @param {type:\"boolean\"}\n",
         "  print_Similarity = True # @param {type:\"boolean\"}\n",
         "  print_Name = True # @param {type:\"boolean\"}\n",
@@ -651,7 +652,8 @@
       ],
       "metadata": {
         "collapsed": true,
-        "id": "fi0jRruI0-tu"
+        "id": "fi0jRruI0-tu",
+        "cellView": "form"
       },
       "execution_count": null,
       "outputs": []