初始化项目,由ModelHub XC社区提供模型
Model: sentence-transformers/gtr-t5-base Source: Original Platform
This commit is contained in:
9
convert_to_fp16.py
Executable file
9
convert_to_fp16.py
Executable file
@@ -0,0 +1,9 @@
|
||||
import sys
|
||||
from transformers import T5EncoderModel
|
||||
|
||||
in_path = sys.argv[1]
|
||||
out_path = sys.argv[2]
|
||||
|
||||
model = T5EncoderModel.from_pretrained(in_path)
|
||||
model.half()
|
||||
model.save_pretrained(out_path)
|
||||
Reference in New Issue
Block a user