abetlen commited on
Commit
7daacef
1 Parent(s): 4242aac

Update convert script

Browse files
Files changed (1) hide show
  1. convert_image_gguf.py +3 -2
convert_image_gguf.py CHANGED
@@ -44,6 +44,7 @@ def main():
44
 
45
  ftype = 1 # fp16
46
 
 
47
  # hidden_size=768,
48
  # intermediate_size=3072,
49
  # projection_dim=512,
@@ -154,11 +155,11 @@ def main():
154
  )
155
 
156
  fout.add_tensor(
157
- "v.sub_GN",
158
  tensors.get_tensor("model.vision_embed_tokens.sub_GN").astype(np.float32),
159
  )
160
  fout.add_tensor(
161
- "v.glb_GN",
162
  tensors.get_tensor("model.vision_embed_tokens.glb_GN").astype(np.float32),
163
  )
164
 
 
44
 
45
  ftype = 1 # fp16
46
 
47
+ # source https://github.com/huggingface/transformers/blob/87134662f73d5e89bb015531ddd1d4662371d317/src/transformers/models/clip/configuration_clip.py#L209
48
  # hidden_size=768,
49
  # intermediate_size=3072,
50
  # projection_dim=512,
 
155
  )
156
 
157
  fout.add_tensor(
158
+ "v.sub_gn",
159
  tensors.get_tensor("model.vision_embed_tokens.sub_GN").astype(np.float32),
160
  )
161
  fout.add_tensor(
162
+ "v.glb_gn",
163
  tensors.get_tensor("model.vision_embed_tokens.glb_GN").astype(np.float32),
164
  )
165