Manli commited on
Commit
d023773
1 Parent(s): 8888e4c

Update model name in the demo notebook

Browse files
Files changed (1) hide show
  1. demo.ipynb +2 -2
demo.ipynb CHANGED
@@ -28,9 +28,9 @@
28
  }
29
  ],
30
  "source": [
31
- "from transformers import AutoModelForVision2Seq, AutoTokenizer, AutoImageProcessor, StoppingCriteria\n",
32
  "import torch\n",
33
- "model_name_or_path = \"Salesforce/xgen-mm-phi3-mini-instruct-r-v1.1\"\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",
 
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",