diff --git a/LLaVA_13b_8bit_colab.ipynb b/LLaVA_13b_8bit_colab.ipynb index 06bc21c..2282d9a 100644 --- 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",