patrickvonplaten's picture
up
724cc13
raw
history blame
233 Bytes
#!/usr/bin/env python3
from transformers import CLIPTextModel
model = CLIPTextModel.from_pretrained("./text_encoder")
model.save_pretrained("./text_encoder", variant="no_ema")
model.save_pretrained("./text_encoder", variant="fp16")