From e56ab52bd25145c8b48885db0d05bce21c51f65f Mon Sep 17 00:00:00 2001 From: eaidova Date: Fri, 3 May 2024 13:20:43 +0400 Subject: [PATCH] fixes --- .../latent-consistency-models-image-generation.ipynb | 4 ---- notebooks/oneformer-segmentation/oneformer-segmentation.ipynb | 3 --- notebooks/sdxl-turbo/sdxl-turbo.ipynb | 2 +- 3 files changed, 1 insertion(+), 8 deletions(-) diff --git a/notebooks/latent-consistency-models-image-generation/latent-consistency-models-image-generation.ipynb b/notebooks/latent-consistency-models-image-generation/latent-consistency-models-image-generation.ipynb index e78afa13eda..50d7bd1eb0c 100644 --- a/notebooks/latent-consistency-models-image-generation/latent-consistency-models-image-generation.ipynb +++ b/notebooks/latent-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 index 87db5a398a3..3b1346073cc 100644 --- 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 index 61944150f5b..0f4900c8025 100644 --- 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",