Update convert script
Browse files- 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.
|
158 |
tensors.get_tensor("model.vision_embed_tokens.sub_GN").astype(np.float32),
|
159 |
)
|
160 |
fout.add_tensor(
|
161 |
-
"v.
|
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 |
|