Ricercar commited on
Commit
6389f31
1 Parent(s): e118181

GemRic-18K clip score fixed!

Browse files
.gitignore CHANGED
@@ -3,4 +3,7 @@ ehthumbs.db
3
  Icon?
4
  Thumbs.db
5
  *.DS_Store
6
- secret.sh
 
 
 
 
3
  Icon?
4
  Thumbs.db
5
  *.DS_Store
6
+
7
+ secret.sh
8
+
9
+ .idea/
Archive/test.py CHANGED
@@ -1,45 +1,124 @@
1
  import streamlit as st
2
  from streamlit_sortables import sort_items
 
 
 
 
 
 
 
 
 
3
 
4
- if __name__ == "__main__":
5
- # if 'check_dict' not in st.session_state:
6
- # st.session_state.check_dict = {'check1': False, 'check2': False, 'check3': False}
7
- #
8
- # with st.form('my_form'):
9
- # st.session_state.check_dict['check1'] = st.checkbox('Check 1 out')
10
- # st.session_state.check_dict['check2'] = st.checkbox('Check 2 out')
11
- # st.session_state.check_dict['check3'] = st.checkbox('Check 3 out')
12
- #
13
- # check21 = st.checkbox('Check 21 out')
14
- # if check21:
15
- # st.write('check21 is checked')
16
- # check22 = st.checkbox('Check 22 out')
17
- # if check22:
18
- # st.write('check22 is checked')
19
- # check23 = st.checkbox('Check 23 out')
20
- # if check23:
21
- # st.write('check23 is checked')
22
- #
23
- # # Every form must have a submit button.
24
- # submitted = st.form_submit_button('Submit')
25
- #
26
- # items = ['a', 'b', 'c']
27
- # sorted_items = sort_items(items)
28
- #
29
- # for key, value in st.session_state.check_dict.items():
30
- # st.write(key, value)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
 
32
- from tqdm import tqdm
33
- import time
34
 
35
- outer_list = [1, 2, 3, 4]
36
- inner_list = [5, 6, 7, 8]
 
37
 
38
- # Outer loop
39
- for item in tqdm(outer_list, desc="Outer Loop", position=0):
40
- # Inner loop with nested=True
41
- for inner_item in tqdm(inner_list, desc="Inner Loop", leave=False, position=1):
42
- # Your nested loop logic here
43
- time.sleep(0.1)
44
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
45
 
 
1
  import streamlit as st
2
  from streamlit_sortables import sort_items
3
+ from torchvision import transforms
4
+ from transformers import CLIPProcessor, CLIPModel
5
+ from torchmetrics.multimodal import CLIPScore
6
+ import torch
7
+ import numpy as np
8
+ import pandas as pd
9
+ from tqdm import tqdm
10
+ from datasets import load_dataset, Dataset, load_from_disk
11
+ import os
12
 
13
+ import clip
14
+
15
+ def compute_clip_score(promptbook, device, drop_negative=False):
16
+ # if 'clip_score' in promptbook.columns:
17
+ # print('==> Skipping CLIP-Score computation')
18
+ # return
19
+ print('==> CLIP-Score computation started')
20
+ clip_scores = []
21
+ to_tensor = transforms.ToTensor()
22
+ # metric = CLIPScore(model_name_or_path='openai/clip-vit-base-patch16').to(DEVICE)
23
+ metric = CLIPScore(model_name_or_path='openai/clip-vit-large-patch14').to(device)
24
+ for i in tqdm(range(0, len(promptbook), BATCH_SIZE)):
25
+ images = []
26
+ prompts = list(promptbook.prompt.values[i:i+BATCH_SIZE])
27
+ for image in promptbook.image.values[i:i+BATCH_SIZE]:
28
+ images.append(to_tensor(image))
29
+ with torch.no_grad():
30
+ x = metric.processor(text=prompts, images=images, return_tensors='pt', padding=True)
31
+ img_features = metric.model.get_image_features(x['pixel_values'].to(device))
32
+ img_features = img_features / img_features.norm(p=2, dim=-1, keepdim=True)
33
+ txt_features = metric.model.get_text_features(x['input_ids'].to(device), x['attention_mask'].to(device))
34
+ txt_features = txt_features / txt_features.norm(p=2, dim=-1, keepdim=True)
35
+ scores = 100 * (img_features * txt_features).sum(axis=-1).detach().cpu()
36
+ if drop_negative:
37
+ scores = torch.max(scores, torch.zeros_like(scores))
38
+ clip_scores += [round(s.item(), 4) for s in scores]
39
+ promptbook['clip_score'] = np.asarray(clip_scores)
40
+ print('==> CLIP-Score computation completed')
41
+ return promptbook
42
+
43
+
44
+ def compute_clip_score_hmd(promptbook):
45
+
46
+ metric_cpu = CLIPScore(model_name_or_path="openai/clip-vit-large-patch14").to('cpu')
47
+ metric_gpu = CLIPScore(model_name_or_path="openai/clip-vit-large-patch14").to('mps')
48
+
49
+ for idx in promptbook.index:
50
+ clip_score_hm = promptbook.loc[idx, 'clip_score']
51
+
52
+ with torch.no_grad():
53
+ image = promptbook.loc[idx, 'image']
54
+ image.save(f"./tmp/{promptbook.loc[idx, 'image_id']}.png")
55
+ image = transforms.ToTensor()(image)
56
+ image_cpu = torch.unsqueeze(image, dim=0).to('cpu')
57
+ image_gpu = torch.unsqueeze(image, dim=0).to('mps')
58
+
59
+ prompts = [promptbook.loc[idx, 'prompt']]
60
+ clip_score_cpu = metric_cpu(image_cpu, prompts)
61
+ clip_score_gpu = metric_gpu(image_gpu, prompts)
62
 
63
+ print(
64
+ f'==> clip_score_hm: {clip_score_hm:.4f}, clip_score_cpu: {clip_score_cpu:.4f}, clip_score_gpu: {clip_score_gpu:.4f}')
65
 
66
+ def compute_clip_score_transformers(promptbook, device='cpu'):
67
+ model = CLIPModel.from_pretrained("openai/clip-vit-large-patch14")
68
+ processor = CLIPProcessor.from_pretrained("openai/clip-vit-large-patch14")
69
 
70
+ with torch.no_grad():
71
+ inputs = processor(text=promptbook.prompt.tolist(), images=promptbook.image.tolist(), return_tensors="pt", padding=True)
72
+ outputs = model(**inputs)
73
+ logits_per_image = outputs.logits_per_image
 
 
74
 
75
+ promptbook.loc[:, 'clip_score'] = logits_per_image[:, 0].tolist()
76
+ return promptbook
77
+
78
+ def compute_clip_score_clip(promptbook, device='cpu'):
79
+ model, preprocess = clip.load("ViT-B/32", device=device)
80
+ with torch.no_grad():
81
+
82
+ for idx in promptbook.index:
83
+ # image_input = preprocess(promptbook.loc[idx, 'image']).unsqueeze(0).to(device)
84
+ image_inputs = preprocess(promptbook.image.tolist()).to(device)
85
+ text_inputs = torch.cat([clip.tokenize(promptbook.prompt.tolist()).to(device)]).to(device)
86
+
87
+ image_features = model.encode_image(image_inputs)
88
+ text_features = model.encode_text(text_inputs)
89
+
90
+
91
+ probs = logits_per_image.softmax(dim=-1).cpu().numpy()
92
+ promptbook.loc[:, 'clip_score'] = probs[:, 0].tolist()
93
+ return promptbook
94
+
95
+
96
+ if __name__ == "__main__":
97
+ BATCH_SIZE = 200
98
+ # DEVICE = 'mps' if torch.has_mps else 'cpu'
99
+
100
+ print(torch.__version__)
101
+
102
+ images_ds = load_from_disk(os.path.join(os.pardir, 'data', 'promptbook'))
103
+ images_ds = images_ds.sort(['prompt_id', 'modelVersion_id'])
104
+ print(images_ds)
105
+ print(type(images_ds[0]['image']))
106
+ promptbook_hmd = pd.DataFrame(images_ds[:20])
107
+ promptbook_new = promptbook_hmd.drop(columns=['clip_score'])
108
+ promptbook_cpu = compute_clip_score(promptbook_new.copy(deep=True), device='cpu')
109
+ promptbook_mps = compute_clip_score(promptbook_new.copy(deep=True), device='mps')
110
+ promptbook_tra_cpu = compute_clip_score_transformers(promptbook_new.copy(deep=True))
111
+ promptbook_tra_mps = compute_clip_score_transformers(promptbook_new.copy(deep=True), device='mps')
112
+ #
113
+ for idx in promptbook_mps.index:
114
+ print(
115
+ 'image id: ', promptbook_mps['image_id'][idx],
116
+ 'mps: ', promptbook_mps['clip_score'][idx],
117
+ 'cpu: ', promptbook_cpu['clip_score'][idx],
118
+ 'tra cpu: ', promptbook_tra_cpu['clip_score'][idx],
119
+ 'tra mps: ', promptbook_tra_mps['clip_score'][idx],
120
+ 'hmd: ', promptbook_hmd['clip_score'][idx]
121
+ )
122
+ #
123
+ # compute_clip_score_hmd(promptbook_hmd)
124
 
data/download_script.py CHANGED
@@ -1,7 +1,9 @@
1
  from datasets import load_dataset, Dataset, load_from_disk
 
2
 
3
 
4
  def main():
 
5
  promptbook = load_dataset('NYUSHPRP/ModelCofferPromptBook', split='train')
6
  print(promptbook)
7
  promptbook.save_to_disk('./promptbook')
 
1
  from datasets import load_dataset, Dataset, load_from_disk
2
+ import os
3
 
4
 
5
  def main():
6
+ os.makedirs('./promptbook', exist_ok=True)
7
  promptbook = load_dataset('NYUSHPRP/ModelCofferPromptBook', split='train')
8
  print(promptbook)
9
  promptbook.save_to_disk('./promptbook')
data/promptbook/data-00000-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:9d4c93a87aefbe7e08ec5ec804054388d0f3df06f6ce211b4d3761904c611367
3
  size 496158744
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9313ff854cf203b11d4ee57c49337daaab1bc1aa137def11a80a2f22247387c8
3
  size 496158744
data/promptbook/data-00001-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bd63d28e10e0b4aa4f9db4120c10f2e4d06bf332cb2df430134cfd121ddfc7a8
3
  size 495185576
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73805e837a90d1878648a43104d73328b553326ea1f00019c954b67ba04ab9ed
3
  size 495185576
data/promptbook/data-00002-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:66f30060bfce10123613b61a8dfbbffa4115a449cf586fff00b06711fdf24e28
3
  size 495550496
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:10252d2c45ea1ba88f3b5e0f3bb0e2b08099606036e6f3638e8449faff36057e
3
  size 495550496
data/promptbook/data-00003-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e4de593cb9a90eee455080a3d6b73aa2dfcd9132e4d87fd81a8a12a73ca4a4dc
3
  size 495623208
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7ff11c7c98b73f2c1824d6f98ea09d8789e6909a728eb39420caa3b8333b2af7
3
  size 495623208
data/promptbook/data-00004-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:92bc097038daf76d4c63021273a7881fcf52d8780a8dd16b77d93ebde71d342b
3
  size 494484472
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4ed29ad5f10230a5f868abd32684db99bb53540036b59cec305f8266b20c48bd
3
  size 494484472
data/promptbook/data-00005-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a7dafef1072812f1e2b70aaf093a04867acc2f841e82c636d3369d05eaa0e00d
3
  size 492936800
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d70104e55ac2752f6dbf658f3619d4dfb55cf9bf958dd422b947bc13370e36c2
3
  size 492936800
data/promptbook/data-00006-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0d1bcd2f094340906d7211ae58d72d708fbd78c3e3d0c4d83de92f5eff52c6a7
3
  size 490180944
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37a485ae5318e30569a98a968115f8907ad0d3e21e1b0f94562c2bde82042e7b
3
  size 490180944
data/promptbook/data-00007-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4f34a697d66716f91ea238f622892f1e9d468991fcae0947b0c4018d674ef421
3
  size 490213760
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dff6f5876530d08a62b0af3b0ff7cce7a141afcb4789cd5aa973ef46ed443680
3
  size 490213760
data/promptbook/data-00008-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4d313cd4d3c449a5161f245c149117a2b4c2506ab8ae4a03a78c8fbd8f1c0aa0
3
  size 492484024
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6c5de841dd2fa29c29acb7b0cfdf9334c61b6d5654bf64ecdd56f0bb5959c956
3
  size 492484024
data/promptbook/data-00009-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3c07c0537f9bb7d2bb22555880fc91e6574a2b2fb320fba26297424c4629cdba
3
  size 494710624
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:de5d0a7132b3befd2c1a94f2ce870d02dffab1c440e65d277e58dc8b40563bf4
3
  size 494710624
data/promptbook/data-00010-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:28a8fdc514f59d42abd692226256b6061256b7cbf6aa43d38c6380c83a469296
3
  size 497841824
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:57cfd4def42baf69d49439021a6d49465df4a2cd2ddb2d8c22170598710f3ea8
3
  size 497841824
data/promptbook/data-00011-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2120bc5f6abed224d236a26397611ba6f1ea4f462b7ab6c8e90ecdbddbd0e3d5
3
  size 494884984
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:35d6f95a91c711453b12f4abf6f67f3bd8cb4b46d02b827f8e62ae0b3c83e4b6
3
  size 494884984
data/promptbook/data-00012-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7dd51105646744c753a5cc0d6ef28520d1343f5456c227685dfb38f9aa4bdcfd
3
  size 488987880
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc94e7858ca01950b0f5a4416a8665e33a9d2428256a81883ef2eadb9458dd3f
3
  size 488987880
data/promptbook/data-00013-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3c9c1e918160b02febc1f8388d469c42bd2c454bebe6e99b681a66afa34a1439
3
  size 498320888
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fcfa4e959819b4766f6085acdb5087be18fbb59e752b9497de7aed1e9eede057
3
  size 498320888
data/promptbook/data-00014-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:325ffa8f6d0f7890e452a142b0049f841a3f19b68a94fa29c558d9bc1c241e3b
3
  size 489513032
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:560c856d65954946f2746b160a7698eac9f01928f8e18d2b126850797fdffe93
3
  size 489513032
data/promptbook/data-00015-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e80dbbaac0f122c08b4b933aeb4c6b93d86e423f935a9395751aecf78278468b
3
  size 495904920
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:31b905f1c84f47005543cbfb4178a0af13d5e066d34a2bc71e3e94823861ed12
3
  size 495904920
data/promptbook/data-00016-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:eb5351757ba8714e2c112eb16b78e906f52ad5c2d91b759672834ef1c3b27b7b
3
  size 490964960
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ab14fdd99b4e3c1a56ff786bfa24d4c679b0973f5d60e93cadf640f03934109d
3
  size 490964960
data/promptbook/data-00017-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:95d4a7b3089beec67d5192b7a567f12e7723757ae0fc10a97e0f3375751af3ef
3
  size 499506568
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a63fa98450893c64027c059a72cb0ef349f237fdb8b6057801899422a1e7f8f
3
  size 499506568
data/promptbook/data-00018-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:19e8351bf7b33e42514d45af4b31ee867f5e991537e9c6f32004280be9439206
3
  size 493819464
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b497f822200a4adc68c2cfb6e92f8de67335d92b3b8b5b7c5dc19f207ec2b79
3
  size 493819464
data/promptbook/data-00019-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:13d51dd80ed80434aa22d5717bfb00d9702538e6224202e31f6d37866f2b110c
3
  size 492267312
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d1c4f49a5fc2f04c9a519669504da3e5462e3bffbc3208c9ec66bb9a11d616b
3
  size 492267312
data/promptbook/data-00020-of-00021.arrow CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:eb046ae2de02d6ba94ad7d405ceb4910fdcb4e5332678d3ddbfd45cfa27fe3d6
3
  size 492600960
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32132052c4f7a2631f3180364d978db9bdc49bd23cfa36493ec42a646960a770
3
  size 492600960
data/promptbook/dataset_info.json CHANGED
@@ -3,92 +3,92 @@
3
  "dataset_size": 10372092350,
4
  "description": "",
5
  "download_checksums": {
6
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00000-of-00021-cb13a9c5601452a8.parquet": {
7
- "num_bytes": 496016415,
8
  "checksum": null
9
  },
10
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00001-of-00021-67e6474368f65d25.parquet": {
11
- "num_bytes": 495045083,
12
  "checksum": null
13
  },
14
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00002-of-00021-05c79adc841f63cc.parquet": {
15
- "num_bytes": 495409904,
16
  "checksum": null
17
  },
18
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00003-of-00021-e97c0ead8bcad17a.parquet": {
19
- "num_bytes": 495478193,
20
  "checksum": null
21
  },
22
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00004-of-00021-a87f2fad83ef0318.parquet": {
23
- "num_bytes": 494335642,
24
  "checksum": null
25
  },
26
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00005-of-00021-d71f50c4c5f247a6.parquet": {
27
- "num_bytes": 492796320,
28
  "checksum": null
29
  },
30
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00006-of-00021-08ac4e752a2fcaa7.parquet": {
31
- "num_bytes": 490037820,
32
  "checksum": null
33
  },
34
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00007-of-00021-3d30edf024e4ccd4.parquet": {
35
- "num_bytes": 490068847,
36
  "checksum": null
37
  },
38
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00008-of-00021-70cd4539c36b75a3.parquet": {
39
- "num_bytes": 492340410,
40
  "checksum": null
41
  },
42
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00009-of-00021-412b974bb974fb8d.parquet": {
43
- "num_bytes": 494567908,
44
  "checksum": null
45
  },
46
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00010-of-00021-b807277448f7c73f.parquet": {
47
- "num_bytes": 497702992,
48
  "checksum": null
49
  },
50
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00011-of-00021-3a68c6a3b4ae4d64.parquet": {
51
- "num_bytes": 494741009,
52
  "checksum": null
53
  },
54
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00012-of-00021-3f3f86b4f5cb17c3.parquet": {
55
- "num_bytes": 488841010,
56
  "checksum": null
57
  },
58
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00013-of-00021-85cb882e4c2dcb86.parquet": {
59
- "num_bytes": 498179776,
60
  "checksum": null
61
  },
62
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00014-of-00021-2afe87982509f967.parquet": {
63
- "num_bytes": 489363721,
64
  "checksum": null
65
  },
66
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00015-of-00021-eec7637847985493.parquet": {
67
- "num_bytes": 495756875,
68
  "checksum": null
69
  },
70
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00016-of-00021-d6dea60081409784.parquet": {
71
- "num_bytes": 490821792,
72
  "checksum": null
73
  },
74
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00017-of-00021-5add9142d5ead5f9.parquet": {
75
- "num_bytes": 499362603,
76
  "checksum": null
77
  },
78
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00018-of-00021-39b85ca86366c05f.parquet": {
79
- "num_bytes": 493676633,
80
  "checksum": null
81
  },
82
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00019-of-00021-677afc50ca78013a.parquet": {
83
- "num_bytes": 492125355,
84
  "checksum": null
85
  },
86
- "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/b1796a1297ac1c43eee27f504c7c2f6e86b3b6d0/data/train-00020-of-00021-47bac33a72742d9d.parquet": {
87
- "num_bytes": 492454315,
88
  "checksum": null
89
  }
90
  },
91
- "download_size": 10369122623,
92
  "features": {
93
  "image": {
94
  "_type": "Image"
@@ -172,7 +172,7 @@
172
  },
173
  "homepage": "",
174
  "license": "",
175
- "size_in_bytes": 20741214973,
176
  "splits": {
177
  "train": {
178
  "name": "train",
 
3
  "dataset_size": 10372092350,
4
  "description": "",
5
  "download_checksums": {
6
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00000-of-00021-dce1ab300958e31f.parquet": {
7
+ "num_bytes": 496016461,
8
  "checksum": null
9
  },
10
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00001-of-00021-bc9644c1d3743699.parquet": {
11
+ "num_bytes": 495045000,
12
  "checksum": null
13
  },
14
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00002-of-00021-ea2e66702961212b.parquet": {
15
+ "num_bytes": 495409902,
16
  "checksum": null
17
  },
18
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00003-of-00021-a9bd0a2782c2c69d.parquet": {
19
+ "num_bytes": 495478259,
20
  "checksum": null
21
  },
22
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00004-of-00021-90c88f250e8f6b62.parquet": {
23
+ "num_bytes": 494335691,
24
  "checksum": null
25
  },
26
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00005-of-00021-b2c5536485ce34b3.parquet": {
27
+ "num_bytes": 492796254,
28
  "checksum": null
29
  },
30
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00006-of-00021-971695ae84e53afe.parquet": {
31
+ "num_bytes": 490037891,
32
  "checksum": null
33
  },
34
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00007-of-00021-3b1e0c653bc45819.parquet": {
35
+ "num_bytes": 490068792,
36
  "checksum": null
37
  },
38
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00008-of-00021-e99a99266f821af8.parquet": {
39
+ "num_bytes": 492340436,
40
  "checksum": null
41
  },
42
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00009-of-00021-16fa05c06e124b7d.parquet": {
43
+ "num_bytes": 494567838,
44
  "checksum": null
45
  },
46
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00010-of-00021-9f81d2c90824a45d.parquet": {
47
+ "num_bytes": 497703057,
48
  "checksum": null
49
  },
50
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00011-of-00021-b1abb8fe24b20a1d.parquet": {
51
+ "num_bytes": 494741057,
52
  "checksum": null
53
  },
54
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00012-of-00021-d67c472e42578ac1.parquet": {
55
+ "num_bytes": 488841048,
56
  "checksum": null
57
  },
58
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00013-of-00021-27e4724254d903b4.parquet": {
59
+ "num_bytes": 498179762,
60
  "checksum": null
61
  },
62
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00014-of-00021-72f0d1125f62ff50.parquet": {
63
+ "num_bytes": 489363694,
64
  "checksum": null
65
  },
66
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00015-of-00021-66562f62d5cfd2d7.parquet": {
67
+ "num_bytes": 495756794,
68
  "checksum": null
69
  },
70
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00016-of-00021-741d2101a8ac38d6.parquet": {
71
+ "num_bytes": 490821807,
72
  "checksum": null
73
  },
74
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00017-of-00021-30d22f60bedc5a0a.parquet": {
75
+ "num_bytes": 499362634,
76
  "checksum": null
77
  },
78
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00018-of-00021-d553d608301c44e2.parquet": {
79
+ "num_bytes": 493676628,
80
  "checksum": null
81
  },
82
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00019-of-00021-7d5ce4f46e18cac7.parquet": {
83
+ "num_bytes": 492125301,
84
  "checksum": null
85
  },
86
+ "https://huggingface.co/datasets/NYUSHPRP/ModelCofferPromptBook/resolve/5944fa3c00a863a530b6d74819d9108dac3b0af9/data/train-00020-of-00021-7f5789631bfc4dd7.parquet": {
87
+ "num_bytes": 492454258,
88
  "checksum": null
89
  }
90
  },
91
+ "download_size": 10369122564,
92
  "features": {
93
  "image": {
94
  "_type": "Image"
 
172
  },
173
  "homepage": "",
174
  "license": "",
175
+ "size_in_bytes": 20741214914,
176
  "splits": {
177
  "train": {
178
  "name": "train",
data/promptbook/state.json CHANGED
@@ -64,7 +64,7 @@
64
  "filename": "data-00020-of-00021.arrow"
65
  }
66
  ],
67
- "_fingerprint": "70624c30882c6104",
68
  "_format_columns": null,
69
  "_format_kwargs": {},
70
  "_format_type": null,
 
64
  "filename": "data-00020-of-00021.arrow"
65
  }
66
  ],
67
+ "_fingerprint": "43ce89820094f517",
68
  "_format_columns": null,
69
  "_format_kwargs": {},
70
  "_format_type": null,