nandigama/Hands-On-Large-Language-Models
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Pin revision of model in Chapter 9 (fixes #21)

Browse Source
This commit is contained in:
Maarten Grootendorst
2024-12-10 11:59:48 +01:00
committed by GitHub
parent d9e2491e70 9a8d0d0fbd
commit f27783879b
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
chapter09/Chapter 9 - Multimodal Large Language Models.ipynb
View File

@@ -1900,9 +1900,13 @@
"import torch\n",
"\n",
"# Load processor and main model\n",
"blip_processor = AutoProcessor.from_pretrained(\"Salesforce/blip2-opt-2.7b\")\n",
"blip_processor = AutoProcessor.from_pretrained(\n",
" \"Salesforce/blip2-opt-2.7b\",\n",
" revision=\"51572668da0eb669e01a189dc22abe6088589a24\" # Choose specific model because of: https://huggingface.co/Salesforce/blip2-opt-2.7b/discussions/39\n",
")\n",
"model = Blip2ForConditionalGeneration.from_pretrained(\n",
" \"Salesforce/blip2-opt-2.7b\",\n",
" revision=\"51572668da0eb669e01a189dc22abe6088589a24\",\n",
" torch_dtype=torch.float16\n",
")\n",
"\n",
@@ -2649,7 +2653,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.14"
"version": "3.10.13"
},
"widgets": {
"application/vnd.jupyter.widget-state+json": {