Spaces:
Sleeping
Sleeping
Update inference/inference.ipynb
Browse files
inference/inference.ipynb
CHANGED
@@ -51,7 +51,7 @@
|
|
51 |
}
|
52 |
],
|
53 |
"source": [
|
54 |
-
"model_name = \"
|
55 |
"model = LlavaPhiForCausalLM.from_pretrained(model_name)"
|
56 |
]
|
57 |
},
|
@@ -171,7 +171,7 @@
|
|
171 |
"from conversation import conv_templates, SeparatorStyle\n",
|
172 |
"\n",
|
173 |
"class MultiModalPhi2:\n",
|
174 |
-
" def __init__(self, modelname_or_path=\"
|
175 |
" temperature=0.2,\n",
|
176 |
" max_new_tokens=1024,\n",
|
177 |
" device=\"cuda:0\"):\n",
|
|
|
51 |
}
|
52 |
],
|
53 |
"source": [
|
54 |
+
"model_name = \"Navyabhat/Llava-Phi2\"\n",
|
55 |
"model = LlavaPhiForCausalLM.from_pretrained(model_name)"
|
56 |
]
|
57 |
},
|
|
|
171 |
"from conversation import conv_templates, SeparatorStyle\n",
|
172 |
"\n",
|
173 |
"class MultiModalPhi2:\n",
|
174 |
+
" def __init__(self, modelname_or_path=\"Navyabhat/Llava-Phi2\",\n",
|
175 |
" temperature=0.2,\n",
|
176 |
" max_new_tokens=1024,\n",
|
177 |
" device=\"cuda:0\"):\n",
|