datasetsANDmodels commited on
Commit
4497ced
·
verified ·
1 Parent(s): b6654e7

Upload bin2safetensor.py

Browse files
Files changed (1) hide show
  1. bin2safetensor.py +27 -0
bin2safetensor.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ '''
2
+ from transformers import BlipForConditionalGeneration
3
+
4
+ model = BlipForConditionalGeneration.from_pretrained(
5
+ "im2t",
6
+ local_files_only=True
7
+ )
8
+
9
+ )
10
+
11
+ print("DONE")
12
+
13
+
14
+ '''
15
+ from transformers import BlipForConditionalGeneration
16
+
17
+ model = BlipForConditionalGeneration.from_pretrained(
18
+ "Salesforce/blip-image-captioning-base",
19
+ use_safetensors=True
20
+ )
21
+
22
+ model.save_pretrained("datasetsANDmodels/image2text")
23
+
24
+ model.save_pretrained(
25
+ "im2t",
26
+ safe_serialization=True
27
+ )