Update inference/inference.ipynb

inference/inference.ipynb

@@ -51,7 +51,7 @@
     }
    ],
    "source": [
-    "model_name = \"RaviNaik/Llava-Phi2\"\n",
+    "model_name = \"GunaKoppula/Llava-Phi2\"\n",
     "model = LlavaPhiForCausalLM.from_pretrained(model_name)"
    ]
   },
@@ -171,7 +171,7 @@
     "from conversation import conv_templates, SeparatorStyle\n",
     "\n",
     "class MultiModalPhi2:\n",
-    "    def __init__(self, modelname_or_path=\"RaviNaik/Llava-Phi2\",\n",
+    "    def __init__(self, modelname_or_path=\"GunaKoppula/Llava-Phi2\",\n",
     "                temperature=0.2,\n",
     "                max_new_tokens=1024,\n",
     "                device=\"cuda:0\"):\n",
@@ -366,4 +366,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}