Update model name in the demo notebook
Browse files- demo.ipynb +2 -2
demo.ipynb
CHANGED
@@ -28,9 +28,9 @@
|
|
28 |
}
|
29 |
],
|
30 |
"source": [
|
31 |
-
"from transformers import AutoModelForVision2Seq, AutoTokenizer, AutoImageProcessor
|
32 |
"import torch\n",
|
33 |
-
"model_name_or_path = \"Salesforce/xgen-mm-phi3-mini-instruct-r-v1.
|
34 |
"model = AutoModelForVision2Seq.from_pretrained(model_name_or_path, trust_remote_code=True)\n",
|
35 |
"tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True, use_fast=False, legacy=False)\n",
|
36 |
"image_processor = AutoImageProcessor.from_pretrained(model_name_or_path, trust_remote_code=True)\n",
|
|
|
28 |
}
|
29 |
],
|
30 |
"source": [
|
31 |
+
"from transformers import AutoModelForVision2Seq, AutoTokenizer, AutoImageProcessor\n",
|
32 |
"import torch\n",
|
33 |
+
"model_name_or_path = \"Salesforce/xgen-mm-phi3-mini-instruct-interleave-r-v1.5\"\n",
|
34 |
"model = AutoModelForVision2Seq.from_pretrained(model_name_or_path, trust_remote_code=True)\n",
|
35 |
"tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, trust_remote_code=True, use_fast=False, legacy=False)\n",
|
36 |
"image_processor = AutoImageProcessor.from_pretrained(model_name_or_path, trust_remote_code=True)\n",
|