test

aifrenz · Oct 13, 2023 · 0e38eb4 · 0e38eb4
1 parent 7784493
commit 0e38eb4
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/LLaVA_13b_8bit_colab.ipynb b/LLaVA_13b_8bit_colab.ipynb
@@ -129,12 +129,6 @@
         "#     load_4bit=False\n",
         "# )\n",
         "\n",
-        "vision_tower = model.get_vision_tower()\n",
-        "if not vision_tower.is_loaded:\n",
-        "    vision_tower.load_model()\n",
-        "vision_tower.to(device='cpu', dtype=torch.float16)\n",
-        "image_processor = vision_tower.image_processor\n",
-        "\n",
         "import requests\n",
         "from PIL import Image\n",
         "from io import BytesIO\n",