fixes

openvinotoolkit · May 3, 2024 · e56ab52 · e56ab52
1 parent 20576bd
commit e56ab52
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 8 deletions.
diff --git a/...tent-consistency-models-image-generation/latent-consistency-models-image-generation.ipynb b/...tent-consistency-models-image-generation/latent-consistency-models-image-generation.ipynb
@@ -565,10 +565,6 @@
     "                return_tensors=\"pt\",\n",
     "            )\n",
     "            text_input_ids = text_inputs.input_ids\n",
-    "            untruncated_ids = self.tokenizer(prompt, padding=\"longest\", return_tensors=\"pt\").input_ids\n",
-    "\n",
-    "            if untruncated_ids.shape[-1] >= text_input_ids.shape[-1] and not torch.equal(text_input_ids, untruncated_ids):\n",
-    "                removed_text = self.tokenizer.batch_decode(untruncated_ids[:, self.tokenizer.model_max_length - 1 : -1])\n",
     "\n",
     "            prompt_embeds = self.text_encoder(text_input_ids, share_inputs=True, share_outputs=True)\n",
     "            prompt_embeds = torch.from_numpy(prompt_embeds[0])\n",

diff --git a/notebooks/oneformer-segmentation/oneformer-segmentation.ipynb b/notebooks/oneformer-segmentation/oneformer-segmentation.ipynb
@@ -839,9 +839,6 @@
     "collapsed": false,
     "jupyter": {
      "outputs_hidden": false
-    },
-    "test_replace": {
-    "subset_size=len(coco_dataset),": "subset_size=10,"
     }
    },
    "outputs": [

diff --git a/notebooks/sdxl-turbo/sdxl-turbo.ipynb b/notebooks/sdxl-turbo/sdxl-turbo.ipynb
@@ -599,7 +599,7 @@
    "cell_type": "code",
    "execution_count": 16,
    "id": "6b62f498",
-   "metadata": {"test_replace": {"subset_size=200": "subset_size=10"},
+   "metadata": {"test_replace": {"subset_size=200": "subset_size=10"}},
    "outputs": [],
    "source": [
     "%%skip not $to_quantize.value\n",