pizb commited on
Commit
71d53da
1 Parent(s): 891ef77

wip: add other baseline

Browse files
Finetune_PaliGemma_for_image_description.ipynb ADDED
The diff for this file is too large to render. See raw diff
 
article_base_tutorial.ipynb CHANGED
@@ -254,7 +254,7 @@
254
  "cell_type": "markdown",
255
  "metadata": {},
256
  "source": [
257
- "# Inference for test"
258
  ]
259
  },
260
  {
@@ -262,7 +262,20 @@
262
  "execution_count": null,
263
  "metadata": {},
264
  "outputs": [],
265
- "source": []
 
 
 
 
 
 
 
 
 
 
 
 
 
266
  }
267
  ],
268
  "metadata": {
 
254
  "cell_type": "markdown",
255
  "metadata": {},
256
  "source": [
257
+ "Save Model"
258
  ]
259
  },
260
  {
 
262
  "execution_count": null,
263
  "metadata": {},
264
  "outputs": [],
265
+ "source": [
266
+ "save_path = \"./fine_tuned_model\"\n",
267
+ "model.save_pretrained(save_path)\n",
268
+ "processor.save_pretrained(save_path)\n",
269
+ "\n",
270
+ "print(f\"Model saved locally at {save_path}\")"
271
+ ]
272
+ },
273
+ {
274
+ "cell_type": "markdown",
275
+ "metadata": {},
276
+ "source": [
277
+ "# Inference for test"
278
+ ]
279
  }
280
  ],
281
  "metadata": {
requirements.txt ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ accelerate==0.34.2
2
+ aiohappyeyeballs==2.4.2
3
+ aiohttp==3.10.6
4
+ aiosignal==1.3.1
5
+ appnope==0.1.4
6
+ asttokens==2.4.1
7
+ attrs==24.2.0
8
+ certifi==2024.8.30
9
+ charset-normalizer==3.3.2
10
+ comm==0.2.2
11
+ datasets==3.0.1
12
+ debugpy==1.8.6
13
+ decorator==5.1.1
14
+ dill==0.3.8
15
+ executing==2.1.0
16
+ filelock==3.16.1
17
+ frozenlist==1.4.1
18
+ fsspec==2024.6.1
19
+ huggingface-hub==0.25.1
20
+ idna==3.10
21
+ ipykernel==6.29.5
22
+ ipython==8.27.0
23
+ ipywidgets==8.1.5
24
+ jedi==0.19.1
25
+ Jinja2==3.1.4
26
+ jupyter_client==8.6.3
27
+ jupyter_core==5.7.2
28
+ jupyterlab_widgets==3.0.13
29
+ MarkupSafe==2.1.5
30
+ matplotlib-inline==0.1.7
31
+ mpmath==1.3.0
32
+ multidict==6.1.0
33
+ multiprocess==0.70.16
34
+ nest-asyncio==1.6.0
35
+ networkx==3.3
36
+ numpy==2.1.1
37
+ packaging==24.1
38
+ pandas==2.2.3
39
+ parso==0.8.4
40
+ peft==0.13.0
41
+ pexpect==4.9.0
42
+ pillow==10.4.0
43
+ pip==24.0
44
+ platformdirs==4.3.6
45
+ prompt_toolkit==3.0.48
46
+ psutil==6.0.0
47
+ ptyprocess==0.7.0
48
+ pure_eval==0.2.3
49
+ pyarrow==17.0.0
50
+ Pygments==2.18.0
51
+ python-dateutil==2.9.0.post0
52
+ pytz==2024.2
53
+ PyYAML==6.0.2
54
+ pyzmq==26.2.0
55
+ regex==2024.9.11
56
+ requests==2.32.3
57
+ safetensors==0.4.5
58
+ setuptools==75.1.0
59
+ six==1.16.0
60
+ stack-data==0.6.3
61
+ sympy==1.13.3
62
+ tokenizers==0.20.0
63
+ torch==2.4.1
64
+ tornado==6.4.1
65
+ tqdm==4.66.5
66
+ traitlets==5.14.3
67
+ transformers==4.45.1
68
+ typing_extensions==4.12.2
69
+ tzdata==2024.2
70
+ urllib3==2.2.3
71
+ wcwidth==0.2.13
72
+ widgetsnbextension==4.0.13
73
+ xxhash==3.5.0
74
+ yarl==1.13.0