diff --git a/clip_interrogator.ipynb b/clip_interrogator.ipynb index 733f76e..921bdd1 100644 --- a/clip_interrogator.ipynb +++ b/clip_interrogator.ipynb @@ -28,7 +28,7 @@ }, { "cell_type": "code", - "execution_count": null, + "execution_count": 1, "metadata": { "cellView": "form", "id": "aP9FjmWxtLKJ" @@ -100,6 +100,7 @@ " if clip_model_name != ci.config.clip_model_name:\n", " ci.config.clip_model_name = clip_model_name\n", " ci.load_clip_model()\n", + " ci.config.flavor_intermediate_count = 2048 if clip_model_name == \"ViT-L-14/openai\" else 1024\n", " image = image.convert('RGB')\n", " if mode == 'best':\n", " return ci.interrogate(image, max_flavors=int(best_max_flavors))\n", @@ -111,7 +112,7 @@ }, { "cell_type": "code", - "execution_count": 3, + "execution_count": 4, "metadata": { "cellView": "form", "colab": { @@ -264,7 +265,7 @@ "provenance": [] }, "kernelspec": { - "display_name": "Python 3.8.10 ('ci')", + "display_name": "Python 3.7.15 ('py37')", "language": "python", "name": "python3" }, @@ -278,12 +279,12 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.8.10" + "version": "3.7.15" }, "orig_nbformat": 4, "vscode": { "interpreter": { - "hash": "90daa5087f97972f35e673cab20894a33c1e0ca77092ccdd163e60b53596983a" + "hash": "1f51d5616d3bc2b87a82685314c5be1ec9a49b6e0cb1f707bfa2acb6c45f3e5f" } } }, diff --git a/clip_interrogator/clip_interrogator.py b/clip_interrogator/clip_interrogator.py index 31094ba..da664a3 100644 --- a/clip_interrogator/clip_interrogator.py +++ b/clip_interrogator/clip_interrogator.py @@ -32,7 +32,7 @@ class Config: blip_offload: bool = False # clip settings - clip_model_name: str = 'ViT-H-14/laion2b_s32b_b79k' + clip_model_name: str = 'ViT-L-14/openai' clip_model_path: str = None # interrogator settings